fix

app.py

@@ -2,8 +2,11 @@ import os
 from PIL import Image
 import streamlit
 from pix2tex.cli import LatexOCR
+from munch import Munch
 
-args = {'checkpoint': os.path.realpath(os.path.join(os.path.dirname(__file__), 'checkpoints/weights.pth')), 'no_resize': False}
+args = Munch({'config': 'settings/config.yaml',
+              'checkpoint': os.path.realpath(os.path.join(os.path.dirname(__file__), 'checkpoints/weights.pth')),
+              'no_resize': False})
 model = LatexOCR(args)
 
 if __name__ == '__main__':
@@ -26,12 +29,11 @@ if __name__ == '__main__':
     if streamlit.button('Convert'):
         if uploaded_file is not None and image is not None:
             with streamlit.spinner('Computing'):
-                response = model(image)
-            if response.ok:
-                latex_code = response.json()
-                streamlit.code(latex_code, language='latex')
-                streamlit.markdown(f'$\\displaystyle {latex_code}$')
-            else:
-                streamlit.error(response.text)
+                try:
+                    latex_code = model(image)
+                    streamlit.code(latex_code, language='latex')
+                    streamlit.markdown(f'$\\displaystyle {latex_code}$')
+                except Exception as e:
+                    streamlit.error(e)
         else:
             streamlit.error('Please upload an image.')

requirements.txt

@@ -1,2 +1,3 @@
 pix2tex
 streamlit
+munch