sukhmani1303 commited on
Commit
7c8eb81
·
verified ·
1 Parent(s): a682b41

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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