Create app.py

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from mlx_vlm import load, generate
+
+# 1. This line loads the model from Hugging Face into the Space's memory
+model, processor = load("mlx-community/gemma-4-e4b-it-8bit")
+
+# 2. This is the function that talks to the AI
+def chat_with_ai(image, prompt):
+    # It takes your picture and your question and gives back an answer
+    output = generate(model, processor, image, prompt, max_tokens=100)
+    return output
+
+# 3. This builds the actual website you see
+demo = gr.Interface(
+    fn=chat_with_ai, 
+    inputs=[gr.Image(type="pil"), gr.Textbox(label="Ask about the image")], 
+    outputs="text"
+)
+
+# 4. This starts the website!
+demo.launch()