xuandin commited on
Commit
ed692f5
·
verified ·
1 Parent(s): 69b5fd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- 3-Class Probability: {result['prob3class']:.2f}
491
- 3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
492
- 2-Class Probability: {result['prob2class']:.2f}
 
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: