Update api.py

api.py

@@ -91,10 +91,8 @@ for model_path in models_paths:
     model = tf.saved_model.load(model_path)
     models.append(model)
 
-class PredictionResponse(BaseModel):
-    ODIR-B-2K-5Class-LastTrain-Xception: dict
 
-@app.post("/predict", response_model=PredictionResponse)
+@app.post("/predict")
 async def predict_endpoint(file: UploadFile = File(...)):
 
     contents = await file.read()
@@ -108,7 +106,4 @@ async def predict_endpoint(file: UploadFile = File(...)):
         prediction = predict(image, model, filters[i])
         result_json[models_names[i]] = result(prediction)
 
-    return result_json
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="localhost", port=8000)
+    return result_json