Update app.py

app.py

@@ -143,14 +143,14 @@ def generate_performance_metrics():
 
     with torch.no_grad():
         for batch in val_loader:
-            inputs = tokenizer(email_text, padding=True, truncation=True, max_length=256, return_tensors="pt")
-            inputs = {key: val.to(device) for key, val in inputs.items()}
+            inputs = {key: val.to(device) for key, val in batch.items()}
+            labels = inputs.pop("labels").to(device)  # Extract labels
 
             outputs = model(**inputs)
-            prediction = torch.argmax(outputs.logits, dim=1).item()
+            predictions = torch.argmax(outputs.logits, dim=1)
 
-            y_true.append(label)
-            y_pred.append(prediction)
+            y_true.extend(labels.cpu().numpy())
+            y_pred.extend(predictions.cpu().numpy())
 
     # Compute accuracy and classification report
     accuracy = accuracy_score(y_true, y_pred)
@@ -164,6 +164,7 @@ def generate_performance_metrics():
     }
 
 
+
 # Gradio Interface
 
 def create_interface():