mayzinoo commited on
Commit
9e30f97
·
verified ·
1 Parent(s): 2f520d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -64,7 +64,8 @@ except Exception as e:
64
  # 'google/gemma-2b-it' is a good option.
65
  # Ensure you set up environment variables or secrets for API keys if using paid models.
66
  try:
67
- llm_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
68
  except Exception as e:
69
  print(f"Error loading LLM pipeline: {e}")
70
  llm_pipeline = None # Placeholder
 
64
  # 'google/gemma-2b-it' is a good option.
65
  # Ensure you set up environment variables or secrets for API keys if using paid models.
66
  try:
67
+ # llm_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
68
+ llm_pipeline = pipeline("text-generation", model="google/gemma-2b-it")
69
  except Exception as e:
70
  print(f"Error loading LLM pipeline: {e}")
71
  llm_pipeline = None # Placeholder