app18

app.py

@@ -31,7 +31,7 @@ from pydantic import BaseModel
 import shutil
 
 # Cell 1: Image Classification Model
-image_pipeline = pipeline(task="image-classification", model="rocioadlc/TrashNet_ResNet152V2")
+image_pipeline = pipeline(task="image-classification", model="rocioadlc/EfficientNetV2L_TFM")
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)
@@ -68,11 +68,10 @@ vectordb = Chroma.from_documents(
     persist_directory=persist_directory
 )
 # define retriever
-retriever = vectordb.as_retriever(search_type="mmr")
+retriever = vectordb.as_retriever(search_kwargs={"k": 1}, search_type="mmr")
 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 /
 Answer in the same language of the question /
 Always say "thanks for asking!" at the end of the answer /
 If the context is not relevant, please answer the question by using your own knowledge about the topic.