Update app.py

app.py

@@ -20,7 +20,7 @@ def sentiment_analysis_DB(input):
         input,
         None,
         add_special_tokens=True,
-        max_length=MAX_LEN,
+        max_length=100,
         pad_to_max_length=True,
         return_token_type_ids=True
     )
@@ -30,13 +30,13 @@ def sentiment_analysis_DB(input):
 
     # Assuming model_DB is a PyTorch model
     output = model_DB(ids, mask, token_type_ids)
+    print('Raw output is ', output)
     
-    final_outputs = output[0].item()  # Extract the scalar value
+    sigmoid_output = torch.sigmoid(output)
+    print('Sigmoid output is ', sigmoid_output)
 
-    if final_outputs == True:
-        result = 1
-    else:
-        result = 0
+    # Assuming you want to use a threshold of 0.5
+    result = 1 if sigmoid_output.item() > 0.5 else 0
 
     return result