Update app.py

app.py

@@ -18,23 +18,22 @@ user_input = st.chat_input("Enter a text prompt:")
 client = InferenceClient(token=hf_token)
 
 # Button to trigger the inference
-if st.button("Generate Text"):
-    if user_input:
-        with st.spinner(f"Generating text using {model_id}..."):
+if user_input:
+    with st.spinner(f"Generating text using {model_id}..."):
             # Perform inference using the selected model
-            response = client.chat.completions.create(
-                model=model_id,
-                messages=[
-                    {"role": "system", "content": "You are a question answering assistant."},
-                    {"role": "user", "content": user_input}
-                ],
-                max_tokens=500,
-                stream=False
-            )
-            st.success("Text generated!")
-            st.write(response['choices'][0]['message']['content'])
-    else:
-        st.warning("Please enter a prompt to generate text.")
+        response = client.chat.completions.create(
+            model=model_id,
+            messages=[
+                {"role": "system", "content": "You are a question answering assistant."},
+                {"role": "user", "content": user_input}
+            ],
+            max_tokens=500,
+            stream=False
+        )
+        st.success("Text generated!")
+        st.write(response['choices'][0]['message']['content'])
+else:
+    st.warning("Please enter a prompt to generate text.")