Update app.py

app.py

@@ -15,10 +15,6 @@ MODEL_IDS = {
     "paligemma-3b-ft-widgetcap-waveui-448": "agentsea/paligemma-3b-ft-widgetcap-waveui-448",
     "paligemma-3b-ft-waveui-896": "agentsea/paligemma-3b-ft-waveui-896"
 }
-PROCESSOR_IDS = {
-    "paligemma-3b-ft-widgetcap-waveui-448": "google/paligemma-3b-pt-448",
-    "paligemma-3b-ft-waveui-896": "google/paligemma-3b-pt-896"
-}
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']
 # Device configuration
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -27,7 +23,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 models = {name: PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
           for name, model_id in MODEL_IDS.items()}
 processors = {name: PaliGemmaProcessor.from_pretrained(processor_id)
-              for name, processor_id in PROCESSOR_IDS.items()}
+              for name, processor_id in MODEL_IDS.items()}
 
 ###### Transformers Inference
 @spaces.GPU