badrex commited on
Commit
e74b4ec
·
verified ·
1 Parent(s): 4e5901e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -21,6 +21,10 @@ MODEL_PATH = "badrex/JASRv1.1"
21
  processor = AutoProcessor.from_pretrained(MODEL_PATH)
22
  model = AutoModelForCTC.from_pretrained(MODEL_PATH)
23
 
 
 
 
 
24
  @spaces.GPU()
25
  def process_audio(audio_path):
26
  """Process audio with return the generated respotextnse.
 
21
  processor = AutoProcessor.from_pretrained(MODEL_PATH)
22
  model = AutoModelForCTC.from_pretrained(MODEL_PATH)
23
 
24
+ # move model and processor to device
25
+ model = model.to(device)
26
+ #processor = processor.to(device)
27
+
28
  @spaces.GPU()
29
  def process_audio(audio_path):
30
  """Process audio with return the generated respotextnse.