Update app.py
Browse files
app.py
CHANGED
|
@@ -26,9 +26,10 @@ review = st.text_area("", height=150)
|
|
| 26 |
|
| 27 |
inputs = tokenizer(review, return_tensors="pt", padding=True, truncation=True)
|
| 28 |
outputs = model(**inputs)
|
|
|
|
|
|
|
| 29 |
probabilities = torch.softmax(logits, dim=-1)
|
| 30 |
|
| 31 |
-
logits = outputs.logits
|
| 32 |
predictions = torch.argmax(probabilities, dim=1).item()
|
| 33 |
if st.button("Calculate Sentiment"):
|
| 34 |
|
|
|
|
| 26 |
|
| 27 |
inputs = tokenizer(review, return_tensors="pt", padding=True, truncation=True)
|
| 28 |
outputs = model(**inputs)
|
| 29 |
+
logits = outputs.logits
|
| 30 |
+
|
| 31 |
probabilities = torch.softmax(logits, dim=-1)
|
| 32 |
|
|
|
|
| 33 |
predictions = torch.argmax(probabilities, dim=1).item()
|
| 34 |
if st.button("Calculate Sentiment"):
|
| 35 |
|