Update app.py

app.py

@@ -1,22 +1,22 @@
-from io import BytesIO
 import os
 import sys
 import tempfile
-
-from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import RedirectResponse, StreamingResponse
 import gradio as gr
 import requests
 import uvicorn
+import torc
+import base64
+
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import RedirectResponse, StreamingResponse
 from typing import List
-import torch
 from pdf2image import convert_from_bytes
 from PIL import Image
 from torch.utils.data import DataLoader
 from transformers import AutoProcessor
-import base64
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
+from io import BytesIO
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), './colpali-main')))
 
@@ -46,10 +46,6 @@ mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
 ds = []
 images = []
 
-@app.get("/")
-def read_root():
-    return RedirectResponse(url="/docs")
-
 @app.post("/index")
 async def index(files: List[UploadFile] = File(...)):
     global ds, images