Update app.py

app.py

@@ -19,7 +19,7 @@ def preprocess(image):
 # 예측 함수
 def predict_image(img):
     img = preprocess(img)
-    prediction = model.predict(img)[0][0]
+    prediction = model.predict(img)[0]
     return {'정상': 1-prediction, '감염': prediction}
 
 # 인터페이스 구성