deepfake_detection_uq

Build error

saakshigupta commited on May 15

Commit

5a6166c

verified ·

1 Parent(s): 710e59a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -346,8 +346,8 @@ def load_blip_models():
             original_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
             # Load fine-tuned BLIP model for GradCAM analysis
-            finetuned_processor = BlipProcessor.from_pretrained("saakshigupta/deepfake-blip-large")
-            finetuned_model = BlipForConditionalGeneration.from_pretrained("saakshigupta/deepfake-blip-large")
             return original_processor, original_model, finetuned_processor, finetuned_model
         except Exception as e:

             original_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
             # Load fine-tuned BLIP model for GradCAM analysis
+            finetuned_processor = BlipProcessor.from_pretrained("saakshigupta/gradcam-xception-finetuned")
+            finetuned_model = BlipForConditionalGeneration.from_pretrained("saakshigupta/gradcam-xception-finetuned")
             return original_processor, original_model, finetuned_processor, finetuned_model
         except Exception as e: