MatteoAldovardi commited on
Commit
ea97ef5
·
1 Parent(s): 0891a4a
Files changed (1) hide show
  1. main.py +5 -1
main.py CHANGED
@@ -137,4 +137,8 @@ async def predict_inference(data: InferenceInput):
137
  message=f"Inference successful using {model_name.upper()} model."
138
  )
139
  except Exception as e:
140
- raise HTTPException(status_code=400, detail=f"Inference failed: {str(e)}")
 
 
 
 
 
137
  message=f"Inference successful using {model_name.upper()} model."
138
  )
139
  except Exception as e:
140
+ raise HTTPException(status_code=400, detail=f"Inference failed: {str(e)}")
141
+
142
+
143
+ if __name__ == "__main__":
144
+ uvicorn.run("your_module:app", host="0.0.0.0", port=8000, reload=True)