Update app.py
Browse files
app.py
CHANGED
@@ -112,13 +112,6 @@ groq_llm = ChatGroq(
|
|
112 |
api_key=chat_groq_api
|
113 |
)
|
114 |
|
115 |
-
from transformers import AutoTokenizer
|
116 |
-
#Original Model Name
|
117 |
-
MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.3"
|
118 |
-
|
119 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
120 |
-
|
121 |
-
|
122 |
from huggingface_hub import login
|
123 |
import os
|
124 |
|
@@ -134,7 +127,7 @@ 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 |
|
|
|
112 |
api_key=chat_groq_api
|
113 |
)
|
114 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
115 |
from huggingface_hub import login
|
116 |
import os
|
117 |
|
|
|
127 |
import torch
|
128 |
|
129 |
#MODEL_PATH = "mistralai/Mistral-7B-Instruct-v0.3"
|
130 |
+
MODEL_PATH = "tiiuae/falcon-rw-1b"
|
131 |
|
132 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
133 |
|