Update app.py

app.py

@@ -48,6 +48,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.prompts import PromptTemplate
 from google.generativeai.types.safety_types import HarmBlockThreshold, HarmCategory
 
+
 template="""You are an advanced image captioning AI assistant for surveillance related images.
 Your task is to refine and improve an initial image caption using relevant contextual information provided.
 You will receive two inputs:
@@ -64,16 +65,6 @@ prompt_template = PromptTemplate(
     input_variables=["initial_caption", "context"],
 )
 
-llm = ChatGoogleGenerativeAI(model="gemini-1.5-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,
-        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-
-    },
-      )
-
 
 # Title
 st.title("Image Caption Surveillance")
@@ -103,6 +94,15 @@ if image_url:
         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,
+                             safety_settings={
+        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+
+    },
+      )
         
         prompt=prompt_template.format(initial_caption=initial_caption, context=context)
         response = llm.invoke(prompt)