alex-abb commited on
Commit
5c635d8
1 Parent(s): 497e58e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,9 +1,10 @@
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)
 
1
  import gradio as gr
2
  import spaces
3
+ ## Load model directly
4
+ from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
5
+
6
+ processor = AutoProcessor.from_pretrained("openai/whisper-large-v3")
7
+ model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-large-v3")
8
 
9
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
10
  @spaces.GPU(duration=120)