Sangjun2 commited on
Commit
364867a
Β·
verified Β·
1 Parent(s): 666e023

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)