alex-abb commited on
Commit
497e58e
1 Parent(s): 3b76ba2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -1,11 +1,9 @@
1
  import gradio as gr
2
  import spaces
3
  # Load model directly
4
- from transformers import AutoTokenizer, AutoModelForCausalLM
5
-
6
- tokenizer = AutoTokenizer.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
7
- model = AutoModelForCausalLM.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
8
-
9
 
10
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
11
  @spaces.GPU(duration=120)
 
1
  import gradio as gr
2
  import spaces
3
  # Load model directly
4
+ # Load model directly
5
+ from transformers import AutoModelForCausalLM
6
+ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-vision-128k-instruct", trust_remote_code=True)
 
 
7
 
8
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
9
  @spaces.GPU(duration=120)