Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse
 from transformers import pipeline
 import gradio as gr
@@ -15,18 +15,18 @@ def image_classifier(image):
         results[result['label']] = result['score']
     return results
 
-# Define Gradio Interface
-gr_interface = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="label")
-
 # Define FastAPI app
 app = FastAPI()
 
+# Define Gradio Interface
+gr_interface = gr.Interface(fn=image_classifier, inputs=gr.inputs.Image(), outputs="label")
+
 # Define route for Gradio interface
 @app.get("/")
-async def gr_interface_route():
-    return HTMLResponse(gr_interface.launch(inline=False, inbrowser=True))
+async def gr_interface_route(request: Request):
+    return HTMLResponse(gr_interface.launch(request))
 
-# Expose the FastAPI app using Uvicorn
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+# Expose the FastAPI app using Uvicorn (for local testing)
+# if __name__ == "__main__":
+#     import uvicorn
+#     uvicorn.run(app, host="0.0.0.0", port=8000)