Update app.py

app.py

@@ -1,3 +1,20 @@
 import gradio as gr
 
-gr.load("models/dima806/indian_food_image_detection").launch()
+# Load the pre-trained model from Hugging Face
+model = gr.load("models/dima806/indian_food_image_detection")
+
+# Create a function to classify images using the loaded model
+def classify_image(image):
+    return model(image)
+
+# Create a Gradio interface with a custom title
+iface = gr.Interface(
+    fn=classify_image, 
+    inputs=gr.Image(type="numpy", label="Upload an image"), 
+    outputs=gr.Label(label="Prediction"), 
+    title="CAPSTONE",
+    description="Upload an image of Indian food to detect what it is."
+)
+
+# Launch the Gradio interface
+iface.launch()