TheHuriShow commited on
Commit
08f5251
·
verified ·
1 Parent(s): 69a6292

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ print(f"Index is ready. Total vectors in index: {index.ntotal}")
71
 
72
  # --- 3. SYNTHETIC GENERATION (IMPROVED MODEL AND PROMPT) ---
73
  print("Loading generative model...")
74
- generator = pipeline('text-generation', model='gpt2')
75
 
76
  def get_recommendations_and_generate(query_ingredients, k=3):
77
  # 1. Get Recommendations
 
71
 
72
  # --- 3. SYNTHETIC GENERATION (IMPROVED MODEL AND PROMPT) ---
73
  print("Loading generative model...")
74
+ generator = pipeline('text-generation', model='gpt2-medium')
75
 
76
  def get_recommendations_and_generate(query_ingredients, k=3):
77
  # 1. Get Recommendations