Alexvatti commited on
Commit
bb5e0e8
·
verified ·
1 Parent(s): e69cfce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
15
  question_answering = pipeline("question-answering", model="deepset/roberta-base-squad2", device=device)
16
  code_generation = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=device)
17
  summarization = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
18
- translation = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr", device=device)
 
19
  text_generation = pipeline("text-generation", model="gpt2", device=device)
20
  text_classification = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english", device=device)
21
 
 
15
  question_answering = pipeline("question-answering", model="deepset/roberta-base-squad2", device=device)
16
  code_generation = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=device)
17
  summarization = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
18
+ #translation = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr", device=device)
19
+ translation = pipeline("translation", model="facebook/m2m100_418M", device=device)
20
  text_generation = pipeline("text-generation", model="gpt2", device=device)
21
  text_classification = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english", device=device)
22