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)
@@ -39,9 +39,9 @@ def predict_image(input_img):
 
 image_gradio_app = gr.Interface(
     fn=predict_image,
-    inputs=gr.Image(label="Select hot dog candidate", sources=['upload', 'webcam'], type="pil"),
+    inputs=gr.Image(label="Select waste candidate", sources=['upload', 'webcam'], type="pil"),
     outputs=[gr.Image(label="Processed Image"), gr.Label(label="Result", num_top_classes=2)],
-    title="Hot Hot Dog? Or Not?",
+    title="What kind of waste do you have?",
 )
 
 # Cell 2: Chatbot Model
@@ -50,8 +50,8 @@ loader = PyPDFDirectoryLoader('pdfs')
 data=loader.load()
 # split documents
 text_splitter = RecursiveCharacterTextSplitter(
-    chunk_size=700,
-    chunk_overlap=100,
+    chunk_size=500,
+    chunk_overlap=70,
     length_function=len
 )
 docs = text_splitter.split_documents(data)