Update app.py

app.py

@@ -4,7 +4,6 @@ import requests
 import io
 from PIL import Image
 from transformers import AutoProcessor, Blip2ForConditionalGeneration
-from peft import PeftModel, PeftConfig
 import torch
 
 
@@ -57,16 +56,9 @@ class UI:
 
 class Model:
     def load_model(self):
-        peft_model_id = "Shrey23/Image-Captioning"
-        # config = PeftConfig.from_pretrained(peft_model_id)
         global model
         global processor
-        PATH = "model/"
-        model = Blip2ForConditionalGeneration.from_pretrained(PATH, local_files_only=True, low_cpu_mem_usage=True, torch_dtype=torch.float16) #, device_map="auto", load_in_8bit=True
-        print(-1)
-        model = PeftModel.from_pretrained(model, peft_model_id)
-
-        print(-2)
+        model = Blip2ForConditionalGeneration.from_pretrained("Shrey23/Image-Captioning",  torch_dtype=torch.float16) #, device_map="auto", load_in_8bit=True
         processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
 
 def main():