Update app.py
Browse files
app.py
CHANGED
@@ -133,7 +133,8 @@ else:
|
|
133 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
134 |
import torch
|
135 |
|
136 |
-
MODEL_PATH = "mistralai/Mistral-7B-Instruct-v0.3"
|
|
|
137 |
|
138 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
139 |
|
|
|
133 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
134 |
import torch
|
135 |
|
136 |
+
#MODEL_PATH = "mistralai/Mistral-7B-Instruct-v0.3"
|
137 |
+
MODEL_PATH = "tiiuae/falcon-1b"
|
138 |
|
139 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
140 |
|