Hugging Face's logo

Spaces:
DoAnCK1
/
garbage_classifier
Sleeping

App Files Community
haxerwddle commited on
Commit
6074907
·
1 Parent(s): 56e1436

UI change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ def classify_image(image):
27
 
28
 
29
  # ------------------ LOAD FLAN-T5 ------------------
30
- tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
31
- chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
32
 
33
  def explain_recycling(class_label):
34
  prompt = f"""
 
27
 
28
 
29
  # ------------------ LOAD FLAN-T5 ------------------
30
+ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
31
+ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
32
 
33
  def explain_recycling(class_label):
34
  prompt = f"""