UI change

app.py

@@ -27,8 +27,8 @@ def classify_image(image):
 
 
 # ------------------ LOAD FLAN-T5 ------------------
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
-chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
+tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
+chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
 
 def explain_recycling(class_label):
     prompt = f"""
@@ -66,7 +66,7 @@ def explain_recycling(class_label):
         top_p=0.9,
     )
 
-    return tokenizer.decode(outputs[0]).replace("<pad>", "").replace("</s>", "")
+    return tokenizer.decode(outputs[0])
 
 
 # ------------------ PIPELINE ------------------