Update app.py
Browse files
app.py
CHANGED
|
@@ -50,10 +50,10 @@ db = create_database()
|
|
| 50 |
def load_model():
|
| 51 |
print("test")
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 53 |
-
"
|
| 54 |
)
|
| 55 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
| 56 |
-
"
|
| 57 |
)
|
| 58 |
return tokenizer, model
|
| 59 |
|
|
|
|
| 50 |
def load_model():
|
| 51 |
print("test")
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 53 |
+
"MBZUAI/LaMini-Flan-T5-783M"
|
| 54 |
)
|
| 55 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
| 56 |
+
"MBZUAI//LaMini-Flan-T5-783M"
|
| 57 |
)
|
| 58 |
return tokenizer, model
|
| 59 |
|