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="microsoft/resnet-50")
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)
@@ -121,12 +121,9 @@ def chat_interface(question):
     else:
         return "Unexpected answer format"
 
-chatbot_gradio_app = gr.Interface(
-    fn=chat_interface, 
-    inputs=[
-        gr.inputs.Textbox(lines=1, label="Question"),
-    ], 
-    outputs="text"
+chatbot_gradio_app = gr.ChatInterface(
+    fn=chat_interface,
+    title='Our doubts are traitors, And make us lose the good we oft might win, By fearing to attempt.'
 )
 
 # Combine both interfaces into a single app