BharadhwajS commited on
Commit
9f9420c
·
verified ·
1 Parent(s): d2b5fad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
10
  #repo_id="google/gemma-7b-it",
11
  #repo_id="google/gemma-1.1-7b-it",
12
- repo_id="openai-community/gpt2"
13
  task="text-generation",
14
  max_new_tokens=512,
15
  top_k=5,
 
9
  llm = HuggingFaceEndpoint(
10
  #repo_id="google/gemma-7b-it",
11
  #repo_id="google/gemma-1.1-7b-it",
12
+ repo_id="openai-community/gpt2",
13
  task="text-generation",
14
  max_new_tokens=512,
15
  top_k=5,