cp

app.py

@@ -1,8 +1,17 @@
 import streamlit as st
+from transformers import pipeline
 
-# Title of the application
-st.title('Hello, World!')
+# Load your model; this example uses the GPT-2 model for text generation.
+generator = pipeline('text-generation', model='gpt2')
 
-# A button that when clicked will show a message
-if st.button('Say hello'):
-    st.write('Hello from Streamlit!')
+st.title('Hugging Face Model Integration')
+
+# Text input
+user_input = st.text_input("Type a sentence to complete", "Streamlit is ")
+
+# Generate text button
+if st.button('Generate'):
+    # Generate text
+    result = generator(user_input, max_length=50, num_return_sequences=1)
+    # Display the generated text
+    st.text_area("Generated Text", result[0]['generated_text'], height=150)

requirements.txt

@@ -1 +1,4 @@
 streamlit
+transformers
+torch
+