Spaces:

DoAnCK1
/

garbage_classifier

Sleeping

haxerwddle commited on Dec 13, 2025

Commit

6b69d1d

1 Parent(s): 92ae382

Analysis change

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ pipe = pipeline(
     model=chat_model,
     tokenizer=tokenizer,
     device_map="auto",
-    max_new_tokens=80
 )
 def clean_chat_output(full_text):
@@ -143,7 +143,7 @@ def explain_recycling(class_label):
     outputs = pipe(
         prompt,
-        max_new_tokens=80,
         do_sample=True,
         top_p = 0.9,
         temperature=0.3

     model=chat_model,
     tokenizer=tokenizer,
     device_map="auto",
+    max_new_tokens=8
 )
 def clean_chat_output(full_text):
     outputs = pipe(
         prompt,
+        max_new_tokens=8,
         do_sample=True,
         top_p = 0.9,
         temperature=0.3