Update app.py

app.py

@@ -1,11 +1,27 @@
 import streamlit as st
+from transformers import pipeline
 
 # Title of the web application
-st.title('Simple Echo Application')
+st.title('Chest and Physical Limitations LLM Query')
 
-# User text input
-user_input = st.text_input("Type something here and I'll echo it back!")
+# Initialize the model pipeline
+# Ensure to use the correct model identifier
+model_name = "Abbeite/chest_and_physical_limitations"
+generator = pipeline('text-generation', model=model_name)
+
+# User prompt input
+user_prompt = st.text_area("Enter your prompt here:")
+
+# Button to generate text
+if st.button('Generate'):
+    if user_prompt:
+        # Generate response
+        try:
+            response = generator(user_prompt, max_length=50, clean_up_tokenization_spaces=True)
+            # Display the generated text
+            st.text_area("Response:", value=response[0]['generated_text'], height=250, disabled=True)
+        except Exception as e:
+            st.error(f"Error generating response: {str(e)}")
+    else:
+        st.warning("Please enter a prompt.")
 
-# Displaying the input back to the user
-if user_input:
-    st.write(f'You typed: {user_input}')