Ketan1011 commited on
Commit
2186dd3
1 Parent(s): 0f3fe48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ class Sentence(BaseModel):
13
  sentence: str
14
 
15
 
16
- # model_name = 'Vamsi/T5_Paraphrase_Paws'
17
- model_name = 'google/flan-t5-large'
18
  tokenizer = AutoTokenizer.from_pretrained(model_name,clean_up_tokenization_spaces=True)
19
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
13
  sentence: str
14
 
15
 
16
+ model_name = 'Vamsi/T5_Paraphrase_Paws'
17
+ # model_name = 'google/flan-t5-large'
18
  tokenizer = AutoTokenizer.from_pretrained(model_name,clean_up_tokenization_spaces=True)
19
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")