husseinelsaadi commited on
Commit
efed4ce
·
verified ·
1 Parent(s): 8659080

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
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