Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ os.environ['LANGCHAIN_API_KEY'] = 'lsv2_pt_ce80aac3833643dd893527f566a06bf9_667d
|
|
25 |
|
26 |
@st.cache_resource
|
27 |
def load_model():
|
28 |
-
model_name = "
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
30 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
31 |
return model, tokenizer
|
|
|
25 |
|
26 |
@st.cache_resource
|
27 |
def load_model():
|
28 |
+
model_name = "facebook/blenderbot-400M-distill"
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
30 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
31 |
return model, tokenizer
|