app17

app.py

@@ -31,7 +31,7 @@ from pydantic import BaseModel
 import shutil
 
 # Cell 1: Image Classification Model
-image_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
+image_pipeline = pipeline(task="image-classification", model="rocioadlc/TrashNet_ResNet152V2")
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)
@@ -69,7 +69,7 @@ vectordb = Chroma.from_documents(
 )
 # define retriever
 retriever = vectordb.as_retriever(search_type="mmr")
-template = """
+prompt_template = """
 Your name is AngryGreta and you are a recycling chatbot with the objective to anwer questions from user in English or Spanish /
 Use the following pieces of context to answer the question if the question is related with recycling /
 No more than two chunks of context /
@@ -83,7 +83,7 @@ question: {question}
 """
 
 # Create the chat prompt templates
-system_prompt = SystemMessagePromptTemplate.from_template(template)
+system_prompt = SystemMessagePromptTemplate.from_template(prompt_template)
 qa_prompt = ChatPromptTemplate(
 	messages=[
 		system_prompt,