alex-abb commited on
Commit
cf04b5e
1 Parent(s): e9031f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,11 +2,11 @@ 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("Qwen/Qwen2-72B-Instruct")
9
- model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-72B-Instruct")
10
 
11
  @spaces.GPU(duration=120)
12
 
 
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