Update app.py

app.py

@@ -38,8 +38,8 @@ device = torch.device('cuda' if 'cuda' in device_name and torch.cuda.is_availabl
 print(f"🧠 Using device: {device}")
 
 print("...Loading Grounding DINO model...")
-# --- ⬇️ UPGRADED MODEL ⬇️ ---
-gnd_model_id = "IDEA-Research/grounding-dino-large"
+# --- ⬇️ CORRECTED MODEL ID ⬇️ ---
+gnd_model_id = "IDEA-Research/grounding-dino-base" 
 processor_gnd = AutoProcessor.from_pretrained(gnd_model_id)
 model_gnd = AutoModelForZeroShotObjectDetection.from_pretrained(gnd_model_id).to(device)
 
@@ -302,5 +302,6 @@ def compare_items():
         print(f"❌ Error in /compare: {e}")
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500
+        
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)