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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ 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="google/gemma-2b-it")
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
  except Exception as e:
69
  print(f"Error loading LLM pipeline: {e}")
70
  llm_pipeline = None # Placeholder