Update app.py

app.py

@@ -1,16 +1,16 @@
 import torch
 from transformers import IdeficsForVisionText2Text, AutoProcessor
-from peft import PeftModel, PeftConfig
+#from peft import PeftModel, PeftConfig
 import gradio as gr
 
 peft_model_id = "mrm8488/idefics-9b-ft-describe-diffusion-bf16-adapter"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-config = PeftConfig.from_pretrained(peft_model_id)
-model = IdeficsForVisionText2Text.from_pretrained(config.base_model_name_or_path, torch_dtype=torch.bfloat16)
-model = PeftModel.from_pretrained(model, peft_model_id)
-processor = AutoProcessor.from_pretrained(config.base_model_name_or_path)
+#config = PeftConfig.from_pretrained(peft_model_id)
+model = IdeficsForVisionText2Text.from_pretrained(peft_model_id, torch_dtype=torch.bfloat16)
+#model = PeftModel.from_pretrained(model, peft_model_id)
+processor = AutoProcessor.from_pretrained(peft_model_id)
 model = model.to(device)
 model.eval()