Spaces:
Sleeping
Sleeping
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -40,8 +40,7 @@ def predict(image):
|
|
40 |
class_id = 1 if pred[0][0] > 0.5 else 0
|
41 |
confidence = pred[0][0] if class_id == 1 else 1 - pred[0][0]
|
42 |
|
43 |
-
return f"Classified as: {class_names[class_id]}
|
44 |
-
Confidence: {confidence:.4f}"
|
45 |
except Exception as e:
|
46 |
return f"Error: {str(e)}"
|
47 |
|
|
|
40 |
class_id = 1 if pred[0][0] > 0.5 else 0
|
41 |
confidence = pred[0][0] if class_id == 1 else 1 - pred[0][0]
|
42 |
|
43 |
+
return f"Classified as: {class_names[class_id]}\nConfidence: {confidence:.4f}"
|
|
|
44 |
except Exception as e:
|
45 |
return f"Error: {str(e)}"
|
46 |
|