Update app.py

app.py

@@ -5,15 +5,15 @@ import openai
 openai.api_key = "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"
 
 # Define the code generation function
-def code_generation(message):
+def code_generation(code):
     # Set the model name and parameters
-    model = "nlpxucan/wizardlm-base"
+    model = "openai-codex"
     max_tokens = 100
 
-    # Generate code using the model
+    # Generate code suggestions using the model
     response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=message,
+        engine="davinci-codex",
+        prompt=code,
         max_tokens=max_tokens,
         temperature=0.7,
         top_p=1.0,
@@ -32,13 +32,16 @@ iface = gr.Interface(
     fn=code_generation,
     inputs="text",
     outputs="text",
-    title="Code Generation with WizardLM",
-    description="Enter a message as code prompt to generate code.",
+    title="Code Writing and Debugging",
+    description="Write code and get code suggestions.",
     examples=[
-        ["Create a function to calculate the factorial of a number."],
-        ["Sort a list of integers in ascending order."],
+        ["for i in range(10):", "\tprint(i)"],
+        ["def factorial(n):", "\tif n == 0:", "\t\treturn 1", "\telse:", "\t\treturn n * factorial(n-1)"]
     ],
-    theme="default"
+    allow_flagging=False,
+    layout="vertical",
+    theme="compact",
+    live=True,
 )
 
 # Run the Gradio interface