Update app.py

app.py

@@ -1,31 +1,34 @@
-from flask import Flask, render_template, request, redirect, url_for
-import os
+import gradio as gr
+from transformers import pipeline
+from PIL import Image, ImageDraw
 
-app = Flask(__name__)
-app.config['UPLOAD_FOLDER'] = 'uploads'
+# Hugging Face에서 YOLO 모델을 로드합니다.
+model_name = "jameslahm/yolov10x"
+yolo_model = pipeline('object-detection', model=model_name)
 
-if not os.path.exists(app.config['UPLOAD_FOLDER']):
-    os.makedirs(app.config['UPLOAD_FOLDER'])
+# 객체 감지와 이미지 그리기 함수를 정의합니다.
+def detect_objects(image):
+    # 객체 감지
+    results = yolo_model(image)
 
-@app.route('/')
-def index():
-    return render_template('index.html')
+    # 감지된 객체에 경계 상자 그리기
+    draw = ImageDraw.Draw(image)
+    for obj in results:
+        box = obj['box']
+        draw.rectangle([box['xmin'], box['ymin'], box['xmax'], box['ymax']], outline="red", width=2)
+        draw.text((box['xmin'], box['ymin']), obj['label'], fill="red")
 
-@app.route('/upload', methods=['POST'])
-def upload_file():
-    if 'file' not in request.files:
-        return redirect(request.url)
-    
-    file = request.files['file']
-    
-    if file.filename == '':
-        return redirect(request.url)
-    
-    if file:
-        filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
-        file.save(filepath)
-        # 여기서 파일 처리 로직을 추가할 수 있습니다.
-        return f'파일이 성공적으로 업로드 되었습니다: {filepath}'
+    return results, image
 
-if __name__ == '__main__':
-    app.run(debug=True)
+# Gradio 인터페이스 설정
+iface = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil"),
+    outputs=[gr.JSON(), gr.Image(type="pil")],
+    title="YOLOv10x Object Detection",
+    description="Upload an image to detect objects using the YOLOv10x model. The results are shown in JSON and the image with bounding boxes."
+)
+
+# Gradio 인터페이스 실행
+if __name__ == "__main__":
+    iface.launch()