add T5 code API

app.py

@@ -4,15 +4,25 @@ import requests as req
 
 code_nl = "function for db connection"
 
-CT5_URL = "https://api-inference.huggingface.co/models/deepset/roberta-base-squad2"
+CT5_URL = "https://api-inference.huggingface.co/models/nielsr/codet5-small-code-summarization-ruby"
 CT5_METHOD = 'POST'
+API_URL = CT5_URL
+headers = {"Authorization": "Bearer api_UhCKXKyqxJOpOcbvrZurQFqmVNZRTtxVfl"}
+
+def query(payload):
+	response = req.post(API_URL, headers=headers, json=payload)
+	return response.json()
+
+output = query({"inputs": "The answer to the universe is"})
+
 
 def pygen_func(code_nl):
     inputs = {'code_nl': code_nl}
     payload = json.dumps(inputs)
-    prediction = req.request(CT5_METHOD, CT5_URL, data=payload)
+    # prediction = req.request(CT5_METHOD, CT5_URL, data=payload)
+    prediction = req.request(CT5_METHOD, CT5_URL, json={"inputs": "The answer to the universe is"})
     answer = json.loads(prediction.content.decode("utf-8"))
-    return 'BOOM!'
+    return 'BOOM! ' + output + ' BOOM!!: ' + answer
 
 iface = gr.Interface(pygen_func, 
     [