Update app.py

app.py

@@ -12,7 +12,8 @@ from contextlib import asynccontextmanager
 import logging
 import aiofiles 
 import torch
-from fastapi import FastAPI, HTTPException, Response, StreamingResponse, UploadFile, File, Form
+from fastapi import FastAPI, HTTPException, UploadFile, File, Form, Query
+from fastapi.responses import Response, StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, Field