Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
| 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
|