Update src/streamlit_app.py

src/streamlit_app.py

@@ -20,11 +20,13 @@ if st.button("Send"):
         with st.spinner("Processing natively through the cluster..."):
             try:
                 # 3. Route via internal endpoints rather than requests.post
-                answer = client.text_generation(
-                    prompt=user_input,
+                response = client.chat.completions.create(
                     model="google/flan-t5-small",
-                    max_new_tokens=50
+                    messages=[{"role": "user", "content": user_input}],
+                    max_tokens=50
                 )
+                
+                # 5. Extract output clean string text safely
                 answer = response.choices[0].message.content
                 st.success(answer)