Update app.py

app.py

@@ -20,7 +20,6 @@ from transformers import AutoModelForImageClassification, AutoImageProcessor
 from langchain import HuggingFacePipeline
 
 
-
 def image_to_query(image):
     """
     input: Image
@@ -29,9 +28,6 @@ def image_to_query(image):
 
     output: Query for the LLM
     """
-    #image = Image.open(image)
-    #image = cv2.imread(image)
-    #image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     image =  Image.fromarray(image)
 
     model = AutoModelForImageClassification.from_pretrained("nprasad24/bean_classifier", from_tf=True)
@@ -121,7 +117,6 @@ def generate_response(rag_chain, query):
 
     output: generated response by the llm
     """
-    #return Markdown(rag_chain.invoke(f"{query}"))
     return rag_chain.invoke(f"{query}")
 
 def main(image):
@@ -132,8 +127,6 @@ def main(image):
     output = generate_response(chain, query)
     return output
 
-#main('test2.jpeg')
-
 title = "Professor Bean: The Bean Disease Expert"
 description = "Professor Bean is an agricultural expert. He will guide you on how to protect your plants from bean diseases"
 app = gr.Interface(fn=main,