Create app.py

app.py

@@ -0,0 +1,29 @@
+import streamlit as st
+from transformers import pipeline
+
+@st.cache(allow_output_mutation=True)
+def load_model():
+    return pipeline("text-generation", model="pbjs_gpt2")
+
+def main():
+    st.title("Hugging Face Model Demo")
+    st.write("Enter some text and get generated text as output.")
+
+    # Create a text input field for user input
+    user_input = st.text_input("Enter text:", "")
+
+    # Check if the user input is empty
+    if user_input:
+        # Load the Hugging Face model
+        generator = load_model()
+
+        # Generate text based on user input
+        generated_text = generator(user_input, max_length=100, num_return_sequences=1)[0]["generated_text"]
+
+        # Display the generated text
+        st.write("Generated Text:")
+        st.write(generated_text)
+
+# Run the app
+if __name__ == "__main__":
+    main()