Update app.py

app.py

@@ -75,8 +75,9 @@ if uploaded_image is not None:
     predicted_class_index = torch.argmax(predictions).item()
     predicted_class = class_names[predicted_class_index]
 
-    val = torch.max(predictions):.2f
-    text = "cat"
+    val = torch.max(predictions)
     if val > 0.5:
         text = "dog"
+    else:
+        text = "cat"
     st.write(f"Prediction: {text} with confidence {val:.2f}")