pgurazada1 commited on
Commit
ee98f4c
·
verified ·
1 Parent(s): a4f9be0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -234,7 +234,9 @@ def formatter(state: State):
234
  return {'final_output': final_output.content}
235
 
236
 
237
- def claim_generator(input_bill_path):
 
 
238
 
239
  workflow = StateGraph(State)
240
 
@@ -252,7 +254,7 @@ def claim_generator(input_bill_path):
252
  workflow.add_edge("formatter", END)
253
 
254
  chain = workflow.compile()
255
-
256
  output = chain.invoke({'image_path': input_bill_path})
257
 
258
  return output['final_output']
 
234
  return {'final_output': final_output.content}
235
 
236
 
237
+ def claim_generator(input_bill_path, progress=gr.Progess()):
238
+
239
+ progress(0, desc="Starting workflow")
240
 
241
  workflow = StateGraph(State)
242
 
 
254
  workflow.add_edge("formatter", END)
255
 
256
  chain = workflow.compile()
257
+ progress(0.05)
258
  output = chain.invoke({'image_path': input_bill_path})
259
 
260
  return output['final_output']