Spaces:
Runtime error
Runtime error
Commit
·
68f4d46
1
Parent(s):
946197d
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
20 |
# st.write(device_map)
|
21 |
|
22 |
@st.cache_resource
|
23 |
-
def load_model()
|
24 |
st.write('Loading model...')
|
25 |
model = pipeline("text-generation", model="bryanmildort/gpt_neo_notes")
|
26 |
st.write('Success!')
|
|
|
20 |
# st.write(device_map)
|
21 |
|
22 |
@st.cache_resource
|
23 |
+
def load_model():
|
24 |
st.write('Loading model...')
|
25 |
model = pipeline("text-generation", model="bryanmildort/gpt_neo_notes")
|
26 |
st.write('Success!')
|