Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def scrape(url: str = Query(...)):
|
|
| 35 |
return ThreadResponse(question="", replies=[])
|
| 36 |
|
| 37 |
|
| 38 |
-
MODEL_NAME = "
|
| 39 |
|
| 40 |
# Load tokenizer and model once at startup, with device auto-mapping
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
|
|
|
| 35 |
return ThreadResponse(question="", replies=[])
|
| 36 |
|
| 37 |
|
| 38 |
+
MODEL_NAME = "google/flan-t5-small"
|
| 39 |
|
| 40 |
# Load tokenizer and model once at startup, with device auto-mapping
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|