Update llm.py

llm.py

@@ -1,22 +1,16 @@
 import pytesseract
 
-import google.generativeai as palm
+import google.generativeai as genai
 api_key = 'AIzaSyB7-RzBwTAfVA-7ZGk2mEOQwOxshpwzhpM' # put your API key here
 palm.configure(api_key=api_key)
-models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods]
-model = models[0].name
+geminiModel = genai.GenerativeModel(model_name='gemini-pro')
+
 
 def llm(img):
     text = pytesseract.image_to_string(img, lang='vie')
     # generate text
-    prompt = "take this peace of information and give all the information in point wise better format also give some recomendation related to them, if you don't get any nutrition content simply reply 'I don't seem have any knowledge of the perticular Nutrition Content' " + text
+    prompt = "take this peace of information and give all the information in point wise better format also give some recomendation related to them' " + text
     # print(prompt)
-    text = palm.generate_text(
-        prompt=prompt,
-        model=model,
-        temperature=0.2,
-        max_output_tokens=2000,
-        top_p=0.8,
-        top_k=40,
-    )
-    return text.result
+    response  = model.generate_content(prompt)
+        
+    return response.text