Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -59,6 +59,7 @@ def format_output(prediction):
|
|
59 |
return prediction.replace('<0x0A>', '\n')
|
60 |
|
61 |
# First model prediction: ko-deplot
|
|
|
62 |
def predict_model1(image):
|
63 |
images = [image]
|
64 |
inputs = processor1(images=images, text="What is the title of the chart", return_tensors="pt", padding=True)
|
|
|
59 |
return prediction.replace('<0x0A>', '\n')
|
60 |
|
61 |
# First model prediction: ko-deplot
|
62 |
+
@spaces.GPU(enable_queue=True,duration=100)
|
63 |
def predict_model1(image):
|
64 |
images = [image]
|
65 |
inputs = processor1(images=images, text="What is the title of the chart", return_tensors="pt", padding=True)
|