changed the way program starts

Dockerfile

@@ -36,4 +36,4 @@ WORKDIR /app
 
 EXPOSE 7860
 # Run the script
-CMD ["uvicorn", "main:app", "--port", "7860"]
+CMD ["python", "main.py"]

main.py

@@ -1,3 +1,4 @@
+import uvicorn
 from fastapi import FastAPI, BackgroundTasks
 from fastapi.responses import FileResponse
 from pydantic import BaseModel
@@ -81,3 +82,6 @@ def download_results(task_id: str):
         shutil.make_archive(folder_path, 'zip', folder_path)
     
     return FileResponse(zip_path, filename=f"{task_id}.zip", media_type="application/zip")
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)