updated output format

app.py

@@ -46,7 +46,8 @@ def predict(image):
     predicted_class = class_names[predicted.item()]
     
     # Format output
-    return f"Predicted class: {predicted_class} (class number: {predicted.item()})"
+    return f"Predicted class: {predicted_class}"
+    return f"Class number: {predicted.item()}"
 
 # Example images from Hugging Face
 examples = [