Update app.py

app.py

@@ -8,6 +8,9 @@ COLORS = [[0.000, 0.447, 0.741], [0.850, 0.325, 0.098], [0.929, 0.694, 0.125], [
 
 import io
 
+processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+
 def fig2img(fig):
     buf = io.BytesIO()
     fig.savefig(buf)
@@ -31,8 +34,6 @@ def plot_results(image, results):
     return fig2img(plt.gcf())
 
 def predict(input_img):
-    processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
-    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
     inputs = processor(images=input_img, return_tensors="pt")
     outputs = model(**inputs)