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

Update app.py

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