Update app.py

app.py

@@ -2,15 +2,20 @@ import os
 import streamlit as st
 from transformers import pipeline
 
+
 st.title("Testing Gemma")
-input_text = st.text_input("How can i help you: ")
-process = st.button("ask")
+input_text = st.text_input("How can I help you: ")
+process = st.button("Ask")
 
 messages = [
     {"role": "user", "content": input_text},
     {"role": "system", "content": "You are a helpful assistant."}
 ]
-pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))
+pipe = pipeline("text-generation", model="google/gemma-3-1b-it", token=os.getenv("HF_TOKEN"))
 
-if process:
-    st.write(pipe(input_text))
+if process and input_text:  
+    output = pipe(input_text)
+    if output and isinstance(output, list) and len(output) > 0 and "generated_text" in output[0]:
+        st.write(output[0]["generated_text"])
+    else:
+        st.write("No response generated or unexpected output format.")