Update app.py

app.py

@@ -7,16 +7,17 @@ from matplotlib import cm
 import torch
 from PIL import Image
 import requests
+from io import BytesIO
 import gradio as gr
 
+processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
+model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 
 def replace(text):
-  processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
-  model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
   # image = Image.open(text).convert("RGB")
   inputs = processor(text, return_tensors="pt")
   outputs = model(**inputs)
-  prediction = processor.post_process_panoptic_segmentation(outputs, target_sizes=[image.size[::-1]])[0]
+  prediction = processor.post_process_panoptic_segmentation(outputs, target_sizes=[text.size[::-1]])[0]
  
   return draw_panoptic_segmentation(**prediction)
 def draw_panoptic_segmentation(segmentation, segments_info):
@@ -56,4 +57,4 @@ interface = gr.Interface(
 )
 
 # Launch the Gradio app
-interface.launch(debug=True)
+interface.launch(debug=True)