0llheaven commited on
Commit
97289f5
·
verified ·
1 Parent(s): 14a095c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
2
  from transformers import AutoImageProcessor, AutoModelForObjectDetection
3
  import torch
4
  from PIL import Image, ImageDraw
5
- import matplotlib.pyplot as plt
6
  import io
7
 
8
  # โหลดโมเดลและตัวประมวลผล
@@ -45,11 +44,11 @@ def detect_objects(image):
45
  # สร้างอินเตอร์เฟซด้วย Gradio
46
  interface = gr.Interface(
47
  fn=detect_objects,
48
- inputs=gr.inputs.Image(type="pil"),
49
- outputs=gr.outputs.Image(type="auto"),
50
  title="Object Detection with Transformers",
51
  description="Upload an image to detect objects using a fine-tuned Conditional-DETR model."
52
  )
53
 
54
  # เปิดใช้งานอินเตอร์เฟซ
55
- interface.launch()
 
2
  from transformers import AutoImageProcessor, AutoModelForObjectDetection
3
  import torch
4
  from PIL import Image, ImageDraw
 
5
  import io
6
 
7
  # โหลดโมเดลและตัวประมวลผล
 
44
  # สร้างอินเตอร์เฟซด้วย Gradio
45
  interface = gr.Interface(
46
  fn=detect_objects,
47
+ inputs=gr.Image(type="pil"),
48
+ outputs=gr.Image(type="auto"),
49
  title="Object Detection with Transformers",
50
  description="Upload an image to detect objects using a fine-tuned Conditional-DETR model."
51
  )
52
 
53
  # เปิดใช้งานอินเตอร์เฟซ
54
+ interface.launch()