Model prompt change

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import random
 import torch
 from transformers import (
     AutoTokenizer, AutoModelForCausalLM,
@@ -36,7 +37,7 @@ pipe = pipeline(
     model=chat_model,
     tokenizer=tokenizer,
     device_map="auto",
-    max_new_tokens=80
+    max_new_tokens=60
 )
 
 def clean_chat_output(full_text):
@@ -75,7 +76,7 @@ def explain_recycling(class_label):
 
     outputs = pipe(
         prompt,
-        max_new_tokens=80,
+        max_new_tokens=60,
         do_sample=True,
         top_p = 0.9,
         temperature=0.3
@@ -87,7 +88,7 @@ def explain_recycling(class_label):
 
 # ------------------ PIPELINE ------------------
 top_label = None
-waste_analization = {
+waste_analyzation = {
     "cardboard": {
         "Recycling type": "Paper/Cardboard recycling",
         "Disposal": "Flatten the cardboard boxes and remove any non-paper packing materials before placing them in the recycling bin.",
@@ -125,6 +126,83 @@ waste_analization = {
         "Extra": "Try to reduce overall waste by reusing and composting when possible."
     }
 }
+waste_analyzation_v2 = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling",
+        "Disposal": "Fold or cut cardboard into smaller flat pieces so it fits properly in recycling bins.",
+        "Tips": "Keep cardboard free from oils or food stains to avoid contaminating the batch.",
+        "Extra": "Cardboard with shiny or laminated surfaces may need separate processing."
+    },
+    "glass": {
+        "Recycling type": "Glass recycling",
+        "Disposal": "Remove lids and caps, lightly rinse, and place glass containers into the glass-only bin.",
+        "Tips": "Do not place broken drinking glasses or ceramics with recyclable glass—they melt at different temperatures.",
+        "Extra": "Some regions require taking glass to drop-off centers instead of curbside bins."
+    },
+    "metal": {
+        "Recycling type": "Metal recycling",
+        "Disposal": "Empty metal containers and place them in the metal bin; labels may stay on.",
+        "Tips": "Keep metal separate from electronics or batteries, which require special facilities.",
+        "Extra": "Large metal objects like appliances may require bulk recycling pickup."
+    },
+    "paper": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Place clean, dry paper products such as envelopes and notebooks into paper bins.",
+        "Tips": "Do not recycle coated paper (e.g., laminated or plastic-lined pages).",
+        "Extra": "Paper with heavy ink coverage might be processed separately."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Clean and drain plastic bottles or tubs, then place them in the plastics bin.",
+        "Tips": "Flatten plastic bottles to reduce volume unless your local guidelines say otherwise.",
+        "Extra": "Some plastics like polystyrene foam require special drop-off locations."
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill",
+        "Disposal": "Dispose of unrecyclable materials in regular waste; secure loose items in bags.",
+        "Tips": "Separate hazardous or toxic waste like paint, solvents, or chemicals.",
+        "Extra": "Reduce landfill impact by choosing reusable products whenever possible."
+    }
+}
+waste_analyzation_v3 = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling",
+        "Disposal": "Remove packing tape when possible and place cardboard in dry storage until pickup day.",
+        "Tips": "Avoid leaving cardboard outdoors where rain could weaken fibers and ruin recyclability.",
+        "Extra": "Food-contaminated cardboard can often be composted instead of recycled."
+    },
+    "glass": {
+        "Recycling type": "Glass recycling",
+        "Disposal": "Sort by color only if required locally; rinse lightly and drop into the proper bin.",
+        "Tips": "Handle carefully to avoid breakage, as shattered glass is often not recyclable in curbside programs.",
+        "Extra": "Glass jars with metal clasps or rubber seals may need partial disassembly before recycling."
+    },
+    "metal": {
+        "Recycling type": "Metal recycling",
+        "Disposal": "Ensure metal food cans are clean and empty before placing in the recycling bin.",
+        "Tips": "Rinse cans briefly—no need for perfect cleaning—as this prevents pests and odors.",
+        "Extra": "Metal lids from jars should be recycled separately from the glass container."
+    },
+    "paper": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Place paper in the appropriate bin; keep shredded paper in a paper bag if accepted.",
+        "Tips": "Avoid mixing paper with wet waste like food scraps to maintain recyclability.",
+        "Extra": "Sticky notes and small scraps may or may not be accepted depending on the facility."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Recycle only plastics accepted by your local program; rinse and drain them completely.",
+        "Tips": "Avoid recycling small plastic pieces under 2 inches—they may jam sorting machinery.",
+        "Extra": "Hard-to-recycle plastics may be collected through special community programs."
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill",
+        "Disposal": "Place all non-recyclables in the trash bin; tie bags securely to avoid leakage.",
+        "Tips": "Do not place batteries, electronics, or sharp objects directly into household trash.",
+        "Extra": "Consider composting organic waste to reduce household trash volume."
+    }
+}
+analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation]
 
 def classify_pipeline(image):
     global top_label
@@ -133,14 +211,15 @@ def classify_pipeline(image):
     top_label = max(predictions, key=predictions.get)  # top-1
     return predictions
 
-def analyze_pipeline():
+def analyze_pipeline(i):
     global top_label
 
     if top_label is None:
         return "Please classify an image first."
     
-    # explanation = explain_recycling(top_label)
-    info = waste_analization[top_label]
+    explanation = explain_recycling(top_label)
+
+    info = analysis[i][top_label]  
     result = (
         f"• Recycling type: {info['Recycling type']}\n"
         f"• Disposal: {info['Disposal']}\n"
@@ -257,11 +336,11 @@ with gr.Blocks() as demo:
     )
 
     # BUTTON
-    analyze_result_btn = gr.Button("Analyze Waste", variant="primary")
+    analyze_result_btn = gr.Button("Analyze", variant="primary")
 
     analyze_result_btn.click(
         analyze_pipeline,
-        inputs=None,
+        inputs=random.randint(0, 2),
         outputs=explain_output
     )