AItool commited on
Commit
f54bb7d
·
verified ·
1 Parent(s): 080e470

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, MarianMTModel, Ma
8
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  MODEL_OPTIONS = [
11
- "Helsinki-NLP", # Round-trip OPUS-MT en→es→en
12
- "FLAN-T5-base (Google en→en)"
13
  ]
14
 
15
  # Cache
 
8
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  MODEL_OPTIONS = [
11
+ "Helsinki-NLP (Acceptable)", # Round-trip OPUS-MT en→es→en
12
+ "FLAN-T5-base (Google poor results)"
13
  ]
14
 
15
  # Cache