Style Changes

core/model_loader.py

@@ -15,12 +15,11 @@ PROMPTS = {
 }
 
 def load_model():
-    # processor = BlipProcessor.from_pretrained(MODEL_ID)
-    # model = BlipForConditionalGeneration.from_pretrained(MODEL_ID)
-    # model.to(DEVICE)
-    # model.eval()
-    # return model, processor
-    return None, None
+    processor = BlipProcessor.from_pretrained(MODEL_ID)
+    model = BlipForConditionalGeneration.from_pretrained(MODEL_ID)
+    model.to(DEVICE)
+    model.eval()
+    return model, processor
 
 def generate_caption(
     model,
@@ -28,7 +27,6 @@ def generate_caption(
     image,
     style
 ):
-    return "Dummay Text"
     prompt = PROMPTS.get(style, "this image shows")
 
     inputs = processor(