fl399 commited on
Commit
50668be
1 Parent(s): ec1ba34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -145,7 +145,7 @@ def get_response_from_openai(prompt, model="gpt-3.5-turbo", max_output_tokens=25
145
  return ret
146
 
147
  ## deplot models
148
- model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16).
149
  if device == "cuda":
150
  model_deplot = model_deplot.to(0)
151
  processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
 
145
  return ret
146
 
147
  ## deplot models
148
+ model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16)
149
  if device == "cuda":
150
  model_deplot = model_deplot.to(0)
151
  processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")