Spaces:
Sleeping
Sleeping
Commit
·
b3fc2f7
1
Parent(s):
46eae05
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
4 |
|
5 |
@st.cache(allow_output_mutation=True)
|
6 |
def load_model():
|
7 |
-
return pipeline("text-generation", model="pbjs_gpt2")
|
8 |
|
9 |
def main():
|
10 |
st.title("Hugging Face Model Demo")
|
|
|
4 |
|
5 |
@st.cache(allow_output_mutation=True)
|
6 |
def load_model():
|
7 |
+
return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
|
8 |
|
9 |
def main():
|
10 |
st.title("Hugging Face Model Demo")
|