Update app.py

app.py

@@ -1,4 +1,3 @@
-from flask import Flask, request, jsonify
 from fastapi import FastAPI
 from pydantic import BaseModel
 import joblib
@@ -150,5 +149,6 @@ def predict(request: URLRequest):
     prediction = get_prediction_from_url(request.url)
     return {"prediction": prediction}
     
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)