Spaces:

unstructuredio
/

receipt-parser

Runtime error

laverdes commited on Jan 6, 2023

Commit

a5dfb39

1 Parent(s): db51e4a

fix: remove admiration mark form Extract all, bug fixed

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ else:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 with col2:
-    if information == 'Extract all!':
         st.text(f'parsing 🧾 (extracting all)...')
         pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
         pretrained_model.to(device)

     device = "cuda" if torch.cuda.is_available() else "cpu"
 with col2:
+    if information == 'Extract all':
         st.text(f'parsing 🧾 (extracting all)...')
         pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
         pretrained_model.to(device)