Update app_dict.py

app_dict.py

@@ -39,10 +39,10 @@ def predict(image):
         print("Raw outputs:", outputs)  # 원본 출력값 확인
         _, predicted = torch.max(outputs, 1)
     label = class_names[predicted.item()]
-    print("Predicted label:", label)
-    
+    if label in ['disposablecup','label'] : result = "Failed"
+    else : result = "Successed (+1)"
     # 예측 결과와 원본 출력값을 모두 반환
-    return label, outputs.numpy()  # 원본 출력값을 NumPy 배열로 변환하여 반환
+    return result  # 원본 출력값을 NumPy 배열로 변환하여 반환
 
 # Gradio 인터페이스 설정 (두 개의 출력)
 iface = gr.Interface(fn=predict, inputs=gr.Image(type='pil'), outputs=['label', 'numpy'], live=True)