alex-abb commited on
Commit
0782b0b
·
verified ·
1 Parent(s): bd2a76b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,11 +2,10 @@ import gradio as gr
2
  import spaces
3
  ## Load model directly
4
  # Load model directly
5
- ## Load model directly
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
8
- tokenizer = AutoTokenizer.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
9
- model = AutoModelForCausalLM.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
10
 
11
  @spaces.GPU(duration=120)
12
 
 
2
  import spaces
3
  ## Load model directly
4
  # Load model directly
 
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
8
+ model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
9
 
10
  @spaces.GPU(duration=120)
11