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")
|