Update app.py

app.py

@@ -92,7 +92,8 @@ def run_inference(image, device, valid_ds):
         predicted_class = torch.argmax(prediction, dim=1).item()
         value_predicted = list(valid_ds.class_to_idx.keys())[list(valid_ds.class_to_idx.values()).index(predicted_class)]
 
-        return f"Predicted Class: {value_predicted}, {predicted_class}"
+        # return f"Predicted Class: {value_predicted}, {predicted_class}"
+        return value_predicted
 
 
     # # Preprocess the image using the feature extractor