Update app.py

app.py

@@ -86,15 +86,15 @@ if image_url:
         logits_per_image = outputs.logits_per_image # this is the image-text similarity score
         probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
         raw_context= scene_labels[probs.argmax(-1)]
-        #context= 'the image is depicting scene of '+ raw_context
-        context= 'the image is depicting scene of attack'
+        context= 'the image is depicting scene of '+ raw_context
+        #context= 'the image is depicting scene of attack'
 
         ##BLIP
         caption = image_to_text(image_url, max_new_tokens=200)
         initial_caption= caption[0]['generated_text']
         ##gemini-1.0-pro-latest
         ##LLM
-        llm = ChatGoogleGenerativeAI(model="gemini-1.0-pro", google_api_key=GOOGLE_API_KEY, temperature=0.2, top_p=1, top_k=1,
+        llm = ChatGoogleGenerativeAI(model="gemini-1.0-pro-latest", google_api_key=GOOGLE_API_KEY, temperature=0.2, top_p=1, top_k=1,
                              safety_settings={
         HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
         HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,