Update app.py

app.py

@@ -166,12 +166,14 @@ llm_model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.4, top_p=0.
 prompt = PromptTemplate(input_variables=['task_type', 'task_number', 'question', 'content', 'description'], template=initial_prompt)
 
 # Define the LLM chain
-chain = LLMChain(
-    llm=llm_model,
-    prompt=prompt,
-)
+chain = LLMChain(llm=llm_model, prompt=prompt)
 
 def evaluate(task_type, task_number, question, image):
+    # Ensure the image is in the correct format
+    if isinstance(image, str):
+        # Load the image if it's a URL or path
+        image = Image.open(image)
+    
     # Process the image to extract text
     text_content = ocr_pipe(image)
     content = text_content[0]['generated_text']
@@ -237,4 +239,4 @@ with gr.Blocks() as demo:
     gr.HTML(footer_html_with_analytics)
 
 # Launch the interface
-demo.launch(share=True, debug=True)
+demo.launch(share=True, debug=True)