Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import gradio as gr
 import transformers
 from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
-from peft import PeftModel
+#from peft import PeftModel
 
 
 if torch.cuda.is_available():
@@ -145,10 +145,9 @@ def get_response_from_openai(prompt, model="gpt-3.5-turbo", max_output_tokens=25
   return ret
 
 ## deplot models
+    model_deplot =  Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16).
 if device == "cuda":
-    model_deplot =  Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16).to(0)
-else: 
-    model_deplot =  Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16)
+    model_deplot = model_deplot.to(0)
 processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
 
 def evaluate(