jdesiree commited on
Commit
a1c7e54
·
verified ·
1 Parent(s): 27e4b47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if "HUGGINGFACEHUB_API_TOKEN" not in os.environ:
18
 
19
  # --- LLM and Template Configuration ---
20
  llm = HuggingFaceEndpoint(
21
- repo_id="google/flan-t5-base",
22
  temperature=0.7,
23
  max_new_tokens=512,
24
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
 
18
 
19
  # --- LLM and Template Configuration ---
20
  llm = HuggingFaceEndpoint(
21
+ repo_id="google/gemma-2b",
22
  temperature=0.7,
23
  max_new_tokens=512,
24
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),