Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
 import torch
 from torchvision import models, transforms
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageDraw
 import numpy as np
-import cv2
 
-model = models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
+model = models.detection.fasterrcnn_resnet50_fpn(weights=models.detection.FasterRCNN_ResNet50_FPN_Weights.COCO_V1)
 model.eval()
 
 transform = transforms.Compose([
@@ -54,15 +53,15 @@ def process_image(image, edit_type, factor):
 iface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.inputs.Image(type="pil"),
-        gr.inputs.Radio(choices=["None", "Crop", "Adjust Color"]),
-        gr.inputs.Slider(0.5, 2.0, step=0.1, label="Factor")
+        gr.Image(type="pil"),
+        gr.Radio(choices=["None", "Crop", "Adjust Color"]),
+        gr.Slider(0.5, 2.0, step=0.1, label="Factor")
     ],
     outputs=[
-        gr.outputs.Image(type="pil"),
-        gr.outputs.Textbox(label="Result")
+        gr.Image(type="pil"),
+        gr.Textbox(label="Result")
     ],
     live=True
 )
 
-iface.launch()
+iface.launch()