changes

app/main.py

@@ -9,24 +9,16 @@ def predict_image(image):
     return pred[0]
 
 
-def create_interface():
-    
+if __name__ == "__main__":
     image_input = gr.Image(type="pil")
-
     output = gr.Text()
 
-    iface = gr.Interface(
+    gr.Interface(
         fn=predict_image,
         inputs=image_input,
         outputs=output,
-        capture_session=True,
         title="Cat or Dog Classifier",
         description="Upload an image to classify whether it's a cat or a dog."
-    )
-    return iface
-
-if __name__ == "__main__":
-    iface = create_interface()
-    iface.launch()
+    ).launch()