BharadhwajS commited on
Commit
67a9023
·
verified ·
1 Parent(s): 946df5e

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="microsoft/phi-2",
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="Falconsai/medical_summarization",
13
  task="text-generation",
14
  max_new_tokens=512,
15
  top_k=5,