Spaces:

PeterBrendan
/

pbjs_gpt2

Sleeping

PeterBrendan commited on Jul 8, 2023

Commit

22c0077

1 Parent(s): d6ce84f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline
 @st.cache(allow_output_mutation=True)
 def load_model():
-    return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
 def main():
     st.title("Hugging Face Model Demo")

 @st.cache(allow_output_mutation=True)
 def load_model():
+    return pipeline("text-generation", model="PeterBrendan/AdsGPT2")
 def main():
     st.title("Hugging Face Model Demo")