update application file

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import AutoProcessor, Gemma3nForConditionalGeneration
-from PIL import Image
 import torch
 import textwrap
 
@@ -17,20 +16,11 @@ model = Gemma3nForConditionalGeneration.from_pretrained(
 def print_response(text: str) -> str:
     return "\n".join(textwrap.fill(line, 100) for line in text.split("\n"))
 
-# 🔍 Inference function
-def predict(image: Image.Image, instruction: str) -> str:
+# 🔍 Inference function for text-only input
+def predict_text(system_prompt: str, user_prompt: str) -> str:
     messages = [
-        {
-            "role": "system",
-            "content": [{"type": "text", "text": "You are a helpful assistant that extracts fields from documents."}],
-        },
-        {
-            "role": "user",
-            "content": [
-                {"type": "image", "image": image},
-                {"type": "text", "text": instruction}
-            ],
-        },
+        {"role": "system", "content": [{"type": "text", "text": system_prompt.strip()}]},
+        {"role": "user", "content": [{"type": "text", "text": user_prompt.strip()}]},
     ]
 
     inputs = processor.apply_chat_template(
@@ -57,14 +47,14 @@ def predict(image: Image.Image, instruction: str) -> str:
 
 # 🎛️ Gradio Interface
 demo = gr.Interface(
-    fn=predict,
+    fn=predict_text,
     inputs=[
-        gr.Image(type="pil", label="Upload Image"),
-        gr.Textbox(lines=2, label="Instruction", value="List visible fields as JSON array {field, value}.")
+        gr.Textbox(lines=2, label="System Prompt", value="You are a helpful assistant."),
+        gr.Textbox(lines=4, label="User Prompt", placeholder="Ask something..."),
     ],
-    outputs=gr.Textbox(label="Output"),
-    title="Gemma 3n Vision + Text",
-    description="Upload an image (e.g., a passport or form) and ask for structured output."
+    outputs=gr.Textbox(label="Gemma 3n Response"),
+    title="Gemma 3n Text-Only Chat",
+    description="Interact with the Gemma 3n language model using plain text. Image input not required.",
 )
 
 if __name__ == "__main__":