Removed file path lowercase conversion and adjusted help text.

Author: R-Haecker

pix2tex.py

@@ -85,7 +85,7 @@ def call_model(args, model, image_resizer, tokenizer):
     else:
         img = np.array(pad(img).convert('RGB'))
         t = test_transform(image=img)['image'][:1].unsqueeze(0)
-
+    print("processing... \n")
     im = t.to(args.device)
 
     with torch.no_grad():
@@ -134,7 +134,8 @@ def call_model(args, model, image_resizer, tokenizer):
     args, *objs = initialize(args)
     while True:
         instructions = input('Predict LaTeX code for image ("?"/"h" for help). ')
-        ins = instructions.strip().lower()
+        possible_file = instructions.strip()
+        ins = possible_file.lower()
         if ins == 'x':
             break
         elif ins in ['?', 'h', 'help']:
@@ -144,16 +145,16 @@ def call_model(args, model, image_resizer, tokenizer):
     On Windows and macOS you can copy the image into memory and just press ENTER to get a prediction.
     Alternatively you can paste the image file path here and submit.
 
-    You might get a different prediction every time you submit the same image. If the result you got was close you \
-can just predict the same image by pressing ENTER again. If that still does not work you can change the temperature \
-or you have take another picture with another resolution.
+    You might get a different prediction every time you submit the same image. If the result you got was close you
+    can just predict the same image by pressing ENTER again. If that still does not work you can change the temperature
+    or you have to take another picture with another resolution (e.g. zoom out and take a screenshot with lower resolution). 
 
     Press "x" to close the program.
     You can interrupt the model if it takes too long by pressing Ctrl+C.
 
 Visualization:
-    You can either render the code into a png using XeLaTeX (see README) to get an image file back. \
-This is slow and requires a working installation of XeLaTeX. To activate type 'show' or set the flag --show
+    You can either render the code into a png using XeLaTeX (see README) to get an image file back.
+    This is slow and requires a working installation of XeLaTeX. To activate type 'show' or set the flag --show
     Alternatively you can render the expression in the browser using katex.org. Type 'katex' or set --katex
 
 Settings:
@@ -165,8 +166,8 @@ def call_model(args, model, image_resizer, tokenizer):
             setattr(args, ins, not getattr(args, ins, False))
             print('set %s to %s' % (ins, getattr(args, ins)))
             continue
-        elif os.path.isfile(ins):
-            args.file = ins
+        elif os.path.isfile(possible_file):
+            args.file = possible_file
         else:
             t = re.match(r't=([\.\d]+)', ins)
             if t is not None: