Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|