Update app.py

app.py

@@ -20,23 +20,29 @@ category_generator = pipeline(
     max_new_tokens=50,
     do_sample=True,
     temperature=1.0
-)
+)  # Flan-T5-small for fast category generation
 analysis_generator = pipeline(
     "text2text-generation",
-    model="google/flan-t5-base",
-    tokenizer="google/flan-t5-base",
-    max_new_tokens=1000,
+    model="google/flan-t5-small",
+    tokenizer="google/flan-t5-small",
+    max_new_tokens=500,  # increased from 200 to 500
     do_sample=True,
     temperature=1.0
-)
+)  # Switched to Flan-T5-small for faster analysis  # reduced tokens for speed
+    do_sample=True,
+    temperature=1.0
+)  # Switched to Flan-T5-small for faster analysis
 suggestion_generator = pipeline(
     "text2text-generation",
-    model="google/flan-t5-base",
-    tokenizer="google/flan-t5-base",
-    max_new_tokens=1000,
+    model="google/flan-t5-small",
+    tokenizer="google/flan-t5-small",
+    max_new_tokens=500,  # increased from 200 to 500
     do_sample=True,
     temperature=1.0
-)
+)  # Using Flan-T5-small for quicker suggestions  # reduced tokens for speed
+    do_sample=True,
+    temperature=1.0
+)  # Using Flan-T5-small for quicker suggestions
 
 # Example URLs for gallery
 def get_recommendations():