Update app.py

app.py

@@ -6,8 +6,8 @@ from PIL import Image
 # .h5 파일을 이용한 모델 로드
 model = tf.keras.models.load_model('keras_model.h5')
 
-# 클래스 레이블 (예: 0은 'cat', 1은 'dog' 등)
-class_labels = ['disposablecup', 'envmark', 'label','mug','nonlabel','reusablecup']  # 모델에 맞게 수정하세요.
+# 클래스 레이블
+class_labels = ['disposablecup', 'envmark', 'label','mug','nonlabel','reusablecup']
 
 # 이미지 전처리 함수
 def preprocess_image(image):
@@ -28,7 +28,7 @@ def predict(image):
 # Gradio 인터페이스 설정
 interface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(source="webcam", tool="editor"),
+    inputs=gr.Image(type="numpy", source="upload", tool="editor"),
     outputs=gr.Textbox(label="Predicted Class"),
     live=True
 )