Update app.py

app.py

@@ -1,12 +1,15 @@
 import gradio as gr
 from transformers import pipeline
 from PIL import Image
+from fastapi import FastAPI
+from starlette.responses import RedirectResponse
 
-# Load summarization and image captioning models
+# Load models
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 image_captioner = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
 
-
+# FastAPI app
+app = FastAPI()
 
 def analyze_input(file, question=None):
     if file is None:
@@ -58,6 +61,7 @@ def analyze_input(file, question=None):
     else:
         return "Unsupported file type. Please upload a valid image or document."
 
+# Gradio Interface
 iface = gr.Interface(
     fn=analyze_input,
     inputs=gr.File(label="Upload Document or Image"),
@@ -66,7 +70,10 @@ iface = gr.Interface(
     description="Upload a document (PDF, DOCX, PPTX, XLSX) to get a summary or an image to get a caption. Runs fully on CPU."
 )
 
-demo = gr.TabbedInterface(iface,"Docs and Images")
+# Wrap in TabbedInterface (even if only one for now)
+demo = gr.TabbedInterface([iface], ["Docs and Images"])
+
+# Mount to FastAPI app
 app = gr.mount_gradio_app(app, demo, path="/")
 
 @app.get("/")