lahiruchamika27 commited on
Commit
bbe537b
·
verified ·
1 Parent(s): 2b24d6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ HF_Token = os.getenv("HF_TOKEN")
7
 
8
  # Initialize the inference client with a coding specialized model
9
  client = InferenceClient(
10
- model="WizardLM/WizardCoder-Python-34B-V1.0", # Using StarCoder2 which excels at code generation
11
  token=HF_Token
12
  )
13
 
 
7
 
8
  # Initialize the inference client with a coding specialized model
9
  client = InferenceClient(
10
+ model="openai-community/gpt2", # Using StarCoder2 which excels at code generation
11
  token=HF_Token
12
  )
13