Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|