Spaces:
Sleeping
Sleeping
Commit
•
c97016b
1
Parent(s):
863194c
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,6 @@ from transformers import pipeline
|
|
| 6 |
import gradio as gr
|
| 7 |
import torch
|
| 8 |
from gtts import gTTS
|
| 9 |
-
import IPython.display as ipd
|
| 10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 11 |
translator_ar_to_en = pipeline("translation_ar_to_en", model="Helsinki-NLP/opus-mt-ar-en", device=0 if device == "cuda" else -1)
|
| 12 |
translator_en_to_ar = pipeline("translation_en_to_arabic", model="Helsinki-NLP/opus-mt-en-ar", device=0 if device == "cuda" else -1)
|
|
|
|
| 6 |
import gradio as gr
|
| 7 |
import torch
|
| 8 |
from gtts import gTTS
|
|
|
|
| 9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 10 |
translator_ar_to_en = pipeline("translation_ar_to_en", model="Helsinki-NLP/opus-mt-ar-en", device=0 if device == "cuda" else -1)
|
| 11 |
translator_en_to_ar = pipeline("translation_en_to_arabic", model="Helsinki-NLP/opus-mt-en-ar", device=0 if device == "cuda" else -1)
|