thushalya commited on
Commit
f9f9f68
1 Parent(s): 403cb04

Update requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -348,6 +348,9 @@ def load_model(tweet):
348
  with torch.no_grad():
349
  model.eval()
350
  pred = model(inputs)
 
 
 
351
  # Assuming your model returns a single value for prediction
352
  pred = torch.round(torch.sigmoid(pred)).item()
353
 
 
348
  with torch.no_grad():
349
  model.eval()
350
  pred = model(inputs)
351
+ print("prediction ",pred)
352
+ print("sigmoid output",torch.sigmoid(pred))
353
+ print("sigmoid item",torch.sigmoid(pred).item())
354
  # Assuming your model returns a single value for prediction
355
  pred = torch.round(torch.sigmoid(pred)).item()
356
 
requirements.txt CHANGED
@@ -49,6 +49,7 @@ Pygments==2.17.2
49
  pyparsing==3.1.2
50
  pyphen==0.15.0
51
  python-dateutil==2.9.0.post0
 
52
  python-multipart==0.0.9
53
  pytz==2024.1
54
  PyYAML==6.0.1
 
49
  pyparsing==3.1.2
50
  pyphen==0.15.0
51
  python-dateutil==2.9.0.post0
52
+ python-dotenv==1.0.1
53
  python-multipart==0.0.9
54
  pytz==2024.1
55
  PyYAML==6.0.1