Update app.py
Browse files
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.
|