updating

nougat_api_app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
-os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"] = "3" # just use one GPU on big machine
+#os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+#os.environ["CUDA_VISIBLE_DEVICES"] = "3" # just use one GPU on big machine
 import torch
 print ('Available devices ', torch.cuda.device_count())
 print ('Current cuda device ', torch.cuda.current_device())
@@ -539,6 +539,13 @@ def read_users(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
     users = crud.get_users(db, skip=skip, limit=limit)
     return users
 
+def stringToBool(string:str) -> bool:
+    if (string.lower()=="true"):
+        return True
+    elif(string.lower()=="false"):
+        return False
+    else: return False
+        
 def sauvolaBinarize(img:Image = None, dest:str = None):
     if dest!= None:
         cv2_image = cv2.imread(dest)
@@ -561,14 +568,7 @@ def sauvolaBinarize(img:Image = None, dest:str = None):
     #dest_filename, dest_ext = os.path.splitext(dest)   
     img.save( './uploads/testBinarization' + ".png", "PNG")   
     return img
-
-def stringToBool(string: str) -> bool:
-    match(string.lower()):
-        case 'true':
-            return True
-        case 'false':
-            return False
-                                
+                               
 @app.post("/predict")
 async def predict(
     request: Request,