Update llm.py

llm.py

@@ -15,11 +15,13 @@ def modelinfo():
     model = [ model for model in palm.list_models() ]
     for m in model:
         print(f" Model Name: {m.name} \n Display Name: {m.display_name} \n Token Limit: {m.input_token_limit} , {m.output_token_limit} \n Temperature: {m.temperature} \n")
+
         
 model = models[0].name
 
 def llm(img):
     text = pytesseract.image_to_string(img, lang='eng')
+    
     # generate text
     prompt = {"take this piece of information and give all the information in point wise better format also give some recommendation related to them, \
                    if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text,
@@ -33,7 +35,7 @@ def llm(img):
                  content and its percentage if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text
                    }
 
-    
+    # Fit model
     text = palm.generate_text(
         prompt=prompt[0],
         model=model,
@@ -42,6 +44,6 @@ def llm(img):
         top_p=0.8,
         top_k=40,
     )
-    return text.result
-
 
+    
+    return text.result