Update app.py

app.py

@@ -98,15 +98,15 @@ async def get_answer_llama(request: Request ):
     text = data['text']
     print("recived ",text)
 
-    try:
-        model = data['model']
-        if model == '13b':
-            print('using 13b')
-            res= do_ML_LLAMA13b(text,0)
-            dict={"LLAMA":res}
-            return JSONResponse(dict)
-    except:
-        print("Using 7b")
+    # try:
+    #     model = data['model']
+    #     if model == '13b':
+    #         print('using 13b')
+    #         res= do_ML_LLAMA13b(text,0)
+    #         dict={"LLAMA":res}
+    #         return JSONResponse(dict)
+    # except:
+    #     print("Using 7b")
         
     
     res= do_ML_LLAMA13b(text,0)
@@ -246,15 +246,15 @@ def do_ML_LLAMA13b(text:str, trycount:int):
                 
                 return "Requested Could not be proceed"
 
-            # try:
-            #     queue_element = driver.find_element(By.CLASS_NAME,'progress-text')
-            #     queue_text = queue_element.text
-            #     print(queue_text)
-            #     if "queue" in queue_text:
-            #         print("Queue is present")
-            #         return do_ML_LLAMA7b(text, trycount)
-            # except:
-            #     pass
+            try:
+                queue_element = driver.find_element(By.CLASS_NAME,'progress-text')
+                queue_text = queue_element.text
+                print(queue_text)
+                if "queue" in queue_text:
+                    print("Queue is present")
+                    return do_ML_LLAMA7b(text, trycount)
+            except:
+                pass
                 
             # value=""
             try: