Update app.py

app.py

@@ -23,13 +23,15 @@ def predict(image):
     predictions = model.predict(processed_image)
     predicted_class_index = np.argmax(predictions, axis=-1)[0]
     predicted_class_label = class_labels[predicted_class_index]
-    return predicted_class_label
+    if predicted_class_label in ['disposablecup','label'] : missionresult = '인증에 실패했습니다.'
+    else : missionresult = '인증에 성공했습니다!'
+    return missionresult
 
 # Gradio 인터페이스 설정
 interface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="numpy"),
-    outputs=gr.Textbox(label="Predicted Class"),
+    outputs=gr.Textbox(label="미션 결과"),
     live=True
 )