Update main.py

main.py

@@ -178,3 +178,6 @@ async def predict(payload: ImagePayload):
 
     except Exception as e:
         return {"error": str(e)}
+
+    if __name__ == "__main__": 
+        import os port = int(os.environ.get("PORT", 8080))