Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -487,9 +487,10 @@ with tabs[0]:
|
|
487 |
gpu_memory_cached = f"{float(final_perf.get('gpu_memory_cached', 0)):.2f} MB" if DEVICE == "cuda" else "N/A"
|
488 |
|
489 |
details = f"""
|
490 |
-
|
491 |
-
3-Class
|
492 |
-
|
|
|
493 |
2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'], int) and result['pred_tc'] != 0 else 'Not used'}
|
494 |
|
495 |
Performance Metrics:
|
|
|
487 |
gpu_memory_cached = f"{float(final_perf.get('gpu_memory_cached', 0)):.2f} MB" if DEVICE == "cuda" else "N/A"
|
488 |
|
489 |
details = f"""
|
490 |
+
Details:
|
491 |
+
- 3-Class Probability: {result['prob3class']:.2f}
|
492 |
+
- 3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
|
493 |
+
- 2-Class Probability: {result['prob2class']:.2f}
|
494 |
2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'], int) and result['pred_tc'] != 0 else 'Not used'}
|
495 |
|
496 |
Performance Metrics:
|