App bug fix.

app.py

@@ -75,10 +75,10 @@ feature_extractor = AutoFeatureExtractor.from_pretrained("hustvl/yolos-small", s
 
 # Build model and load checkpoint
 checkpoint = './checkpoints/epoch=1-step=2184.ckpt'
-model = Detr.load_from_checkpoint(checkpoint, lr=2.5e-5, weight_decay=1e-4)
+model_yolos = Detr.load_from_checkpoint(checkpoint, lr=2.5e-5, weight_decay=1e-4)
 
-model.to(device)
-model.eval()
+model_yolos.to(device)
+model_yolos.eval()
 
 # colors for visualization
 COLORS = [[0.000, 0.447, 0.741], [0.850, 0.325, 0.098], [0.929, 0.694, 0.125],
@@ -134,10 +134,10 @@ def visualize_preds(image, preds, threshold=0.9):
     return plot_results(image, probas[keep], bboxes_scaled)
 
 
-def detect(img, model):
+def detect(img):
 
     # Run inference
-    preds = generate_preds(feature_extractor, model, img)
+    preds = generate_preds(feature_extractor, model_yolos, img)
 
     return visualize_preds(img, preds)