AIPretender commited on
Commit
8d8abfc
·
verified ·
1 Parent(s): 2b14d35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
12
  # repo_id="facebook/bart-large-cnn", # Summarization-capable model
13
  # model_kwargs={"temperature": 0.7, "max_length": 512}
14
  #)
15
- repo_id = "deepseek-ai/DeepSeek-R1-0528"
16
 
17
  llm = HuggingFaceEndpoint(
18
  repo_id=repo_id,
 
12
  # repo_id="facebook/bart-large-cnn", # Summarization-capable model
13
  # model_kwargs={"temperature": 0.7, "max_length": 512}
14
  #)
15
+ repo_id = "google/gemma-2-2b-it"
16
 
17
  llm = HuggingFaceEndpoint(
18
  repo_id=repo_id,