Update app.py

app.py

@@ -22,7 +22,14 @@ if st.button("Generate Text"):
     if user_input:
         with st.spinner(f"Generating text using {model_id}..."):
             # Perform inference using the selected model
-            response = client.text_generation(user_input)
+            response = client.text_generation(model=
+                messages=[
+                    {"role": "system", "content": "You are a question answering assistant."},
+                    {"role": "user", "content": prompt1}
+                ],
+                max_tokens=500,
+                stream=False,
+            ))
             st.success("Text generated!")
             st.write(response)
     else: