molehh commited on
Commit
c76ade6
·
verified ·
1 Parent(s): 58dc04e

changes in main

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -64,7 +64,7 @@ logreg_model.fit(X_train_embeddings, y_train)
64
  class TextInput(BaseModel):
65
  text: str
66
 
67
- @app.post("/predict")
68
  async def generate_prediction(text_input: TextInput):
69
  """
70
  Predict the label for the given text input using the trained model.
 
64
  class TextInput(BaseModel):
65
  text: str
66
 
67
+ @app.post("/prediction")
68
  async def generate_prediction(text_input: TextInput):
69
  """
70
  Predict the label for the given text input using the trained model.