Update llm.py

llm.py

@@ -23,21 +23,21 @@ 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, \
+    prompt = { "0" : "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,
                    
-               "Take this Nutrition facts information and give all the contents and proportion in point wise Markdown format also give some recommendation related to them, \
+               "1" : "Take this Nutrition facts information and give all the contents and proportion in point wise Markdown format also give some recommendation related to them, \
                 Make sure the Recommendations are given in bulleted format under the heading Recommendations \
                    if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text,
 
-                "I've given you this piece of information it contains Nutrition facts, and I want you to give all the information in point-wise Markdown format also give some \
+               "2" : "I've given you this piece of information it contains Nutrition facts, and I want you to give all the information in point-wise Markdown format also give some \
                  recommendation related to them like how the consumption of the content may affect your health and what kind of people will be benefited or harmed from a particular, \
                  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],
+        prompt=prompt["0"],
         model=model,
         temperature=0.3,
         max_output_tokens=2000,
@@ -45,5 +45,4 @@ def llm(img):
         top_k=40,
     )
 
-    
     return text.result