Sangjun2 commited on
Commit
3661fce
Β·
verified Β·
1 Parent(s): 364867a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def format_output(prediction):
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)
66
  inputs = {k: v.to(device) for k, v in inputs.items()} # Move to GPU
67
 
68
  model1.eval()
 
62
  @spaces.GPU(enable_queue=True,duration=100)
63
  def predict_model1(image):
64
  images = [image]
65
+ inputs = processor1(images=images, text="Generate underlying data table of the figure below:", return_tensors="pt", padding=True)
66
  inputs = {k: v.to(device) for k, v in inputs.items()} # Move to GPU
67
 
68
  model1.eval()