Hebrew Poetry - GPT Neo (Small)

README.md

@@ -1,6 +1,6 @@
 ---
-title: Hebrew Poetry - GPT Neo (XL)
-emoji: 🌵
+title: Hebrew Poetry - GPT Neo (Small)
+emoji: 📚
 colorFrom: blue
 colorTo: gray
 sdk: streamlit

app.py

@@ -74,9 +74,9 @@ def extend(input_text, max_size=20, top_k=50, top_p=0.95):
 
 
 if __name__ == "__main__":
-    st.title("Hebrew Poetry - GPT Neo (XL)")
+    st.title("Hebrew Poetry - GPT Neo (Small)")
 
-    model, tokenizer = load_model("Norod78/hebrew-gpt_neo-xl-poetry")
+    model, tokenizer = load_model("Norod78/hebrew-gpt_neo-small")
     #model, tokenizer = load_model("Norod78/hebrew_poetry-gpt_neo-tiny")
 
     stop_token = "<|endoftext|>"