Update app.py

app.py

@@ -6,6 +6,7 @@ from fastapi import FastAPI, HTTPException
 from PIL import Image
 from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import uvicorn 
 
 app = FastAPI()
 
@@ -54,3 +55,6 @@ def query(data: RequestData):
             return {"response": str(response)}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+
+if __name__ == "__main__":
+    uvicorn.run(app:app)