Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ os.environ['LANGCHAIN_API_KEY'] = 'lsv2_pt_ce80aac3833643dd893527f566a06bf9_667d
|
|
23 |
|
24 |
@st.cache_resource
|
25 |
def load_model():
|
26 |
-
model_name = "bigscience/bloom-560m
|
27 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
28 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
29 |
return model, tokenizer
|
|
|
23 |
|
24 |
@st.cache_resource
|
25 |
def load_model():
|
26 |
+
model_name = "bigscience/bloom-560m"
|
27 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
28 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
29 |
return model, tokenizer
|