Update app.py

app.py

@@ -6,13 +6,18 @@ from color_wheel import ColorWheel
 from PIL import ImageDraw, ImageFont
 import numpy as np
 
+resnet_101_processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-101')
+resnet_101_model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-101')
+resnet_50_processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-50')
+resnet_50_model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-50')
+
 def process_image(image, margin, model):
 	if model=='detr-resnet-101':
-		processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-101')
-		model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-101')
+		processor = resnet_101_processor
+		model = resnet_101_model
 	else:
-		processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-50')
-		model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-50')
+		processor = resnet_50_processor
+		model = resnet_50_model
 
 	if image is None:
 		yield [None, None, None]