thushalya
commited on
Commit
•
f9f9f68
1
Parent(s):
403cb04
Update requirements.txt
Browse files- app.py +3 -0
- 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
|