acecalisto3 commited on
Commit
ff07664
·
verified ·
1 Parent(s): 7479428

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -118,7 +118,7 @@ plugin_manager = PluginManager()
118
  plugin_manager.load_plugins()
119
 
120
  # AI Assistant
121
- model = AutoModelForSequenceClassification.from_pretrained("EleutherAI/code-davinci-002")
122
  codex_pipeline = pipeline("code-generation", model=model)
123
 
124
  hf_api = HfApi()
 
118
  plugin_manager.load_plugins()
119
 
120
  # AI Assistant
121
+ model = AutoModelForSequenceClassification.from_pretrained("microsoft/CodeGPT-small-py")
122
  codex_pipeline = pipeline("code-generation", model=model)
123
 
124
  hf_api = HfApi()