Spaces:
Runtime error
Runtime error
changes in main
Browse files
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("/
|
| 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.
|