Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
-from transformers import IdeficsForVisionText2Text, AutoProcessor
+from transformers import AutoProcessor
+#from transformers import IdeficsForVisionText2Text, AutoProcessor
 #from peft import PeftModel, PeftConfig
 import gradio as gr
 
@@ -9,7 +10,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 #config = PeftConfig.from_pretrained(peft_model_id)
-model = IdeficsForVisionText2Text.from_pretrained(peft_model_id, torch_dtype=torch.bfloat16)
+model = AutoProcessor.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)