stmnk commited on
Commit
c9893a4
·
1 Parent(s): 446109d
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import requests as req
4
 
5
  code_nl = "function for db connection"
6
 
7
- CT5_URL = "https://api-inference.huggingface.co/models/nielsr/codet5-small-code-summarization-ruby"
 
8
  CT5_METHOD = 'POST'
9
  API_URL = CT5_URL
10
  headers = {"Authorization": "Bearer api_UhCKXKyqxJOpOcbvrZurQFqmVNZRTtxVfl"}
 
4
 
5
  code_nl = "function for db connection"
6
 
7
+ # CT5_URL = "https://api-inference.huggingface.co/models/nielsr/codet5-small-code-summarization-ruby"
8
+ CT5_URL = "https://api-inference.huggingface.co/models/stmnk/codet5-small-code-summarization-python"
9
  CT5_METHOD = 'POST'
10
  API_URL = CT5_URL
11
  headers = {"Authorization": "Bearer api_UhCKXKyqxJOpOcbvrZurQFqmVNZRTtxVfl"}