Spaces:

gyesibiney
/

sentiment-movie-review-FASTAPI-2

Sleeping

gyesibiney commited on Sep 13, 2023

Commit

5f5fd58

1 Parent(s): e8cce98

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -24,7 +24,7 @@ class SentimentRequest(BaseModel):
 # Define a response model
 class SentimentResponse(BaseModel):
-    sentiment: int  # 1 for positive, 0 for negative
     score: float
 # Create an endpoint for sentiment analysis with query parameter

 # Define a response model
 class SentimentResponse(BaseModel):
+    sentiment: str  # 1 for positive, 0 for negative
     score: float
 # Create an endpoint for sentiment analysis with query parameter