Update app.py

app.py

@@ -2,39 +2,38 @@ import requests
 import json
 import gradio as gr
 
-url="https://huggingface.co/curiouscurrent/omnicode"
+url = "https://huggingface.co/curiouscurrent/omnicode"
 
-headers={
-
-    'Content-Type':'application/json'
+headers = {
+    'Content-Type': 'application/json'
 }
 
-history=[]
+history = []
 
 def generate_response(prompt):
+    global history
     history.append(prompt)
-    final_prompt="\n".join(history)
-
-    data={
-        "model":"curiouscurrent/omnicode",
-        "prompt":final_prompt,
-        "stream":False
+    final_prompt = "\n".join(history)
+    data = {
+        "model": "curiouscurrent/omnicode",
+        "prompt": final_prompt,
+        "stream": False
     }
+    response = requests.post(url, headers=headers, data=json.dumps(data))
 
-    response=requests.post(url,headers=headers,data=json.dumps(data))
-
-    if response.status_code==200:
-        response=response.text
-        data=json.loads(response)
-        actual_response=data['response']
+    if response.status_code == 200:
+        response = response.json()
+        actual_response = response['response']
         return actual_response
     else:
-        print("error:",response.text)
+        print("error:", response.text)
 
 
-interface=gr.Interface(
+interface = gr.Interface(
     fn=generate_response,
-    inputs=gr.Textbox(lines=4,placeholder="Enter your Prompt"),
-    outputs="text"
+    inputs=gr.Textbox(lines=4, placeholder="Enter your Prompt"),
+    outputs="text",
+    title="Omnicode",
+    description="A code generation model trained on a variety of programming languages."
 )
-interface.launch()
+interface.launch()