Update app.py
Browse files
app.py
CHANGED
|
@@ -44,6 +44,8 @@ def perform_inference(text, dataset):
|
|
| 44 |
result = pd.DataFrame({
|
| 45 |
'aspect': result[0]['aspect'],
|
| 46 |
'sentiment': result[0]['sentiment'],
|
|
|
|
|
|
|
| 47 |
'position': result[0]['position']
|
| 48 |
})
|
| 49 |
|
|
|
|
| 44 |
result = pd.DataFrame({
|
| 45 |
'aspect': result[0]['aspect'],
|
| 46 |
'sentiment': result[0]['sentiment'],
|
| 47 |
+
# 'probability': result[0]['probs'],
|
| 48 |
+
'confidence': [round(x, 4) for x in result[0]['confidence']],
|
| 49 |
'position': result[0]['position']
|
| 50 |
})
|
| 51 |
|