gouravbhadraDev commited on
Commit
5e20407
·
verified ·
1 Parent(s): 7929343

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -131,8 +131,8 @@ def scrape(url: str = Query(...)):
131
 
132
  # --- Load T5-Small Model and Tokenizer ---
133
 
134
- tokenizer = T5Tokenizer.from_pretrained("google/t5-small")
135
- model = T5ForConditionalGeneration.from_pretrained("google/t5-small")
136
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
137
  model = model.to(device)
138
 
 
131
 
132
  # --- Load T5-Small Model and Tokenizer ---
133
 
134
+ tokenizer = T5Tokenizer.from_pretrained("google-t5/t5-small")
135
+ model = T5ForConditionalGeneration.from_pretrained("google-t5/t5-small")
136
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
137
  model = model.to(device)
138