Update app.py

app.py

@@ -2,25 +2,26 @@ import gradio as gr
 import requests
 import json
 
-URL = "198.175.88.52"
-myport = "8080"
-g2url  = f"http://{URL}:{myport}/generate"
-
-prompt="Why is the sky purple"
-build_curl_prompt="curl 127.0.0.1:8080/generate -X POST -d '{\"inputs\":\"${prompt}\",\"parameters\":{\"max_new_tokens\":32}}' -H 'Content-Type: application/json'"
-
-url_input = gr.inputs.Textbox(label="URL", value=g2url, visible=False)
-prompt_input = gr.inputs.Textbox(label="Prompt", value=prompt, visible=True)
-outputs = gr.outputs.Textbox(label="Generated Text")
-
-def text_gen(url, prompt):
-    resp = requests.post(url, data=json.dumps(prompt))
-    return resp.text
+def text_gen(prompt, url):
+    data = {
+        "inputs": prompt
+    }
+    try:
+        response = requests.post(url, data=json.dumps(data), headers={'Content-Type': 'application/json'})
+        response.raise_for_status()
+        return response.text
+    except requests.exceptions.RequestException as e:
+        return f"Error: {e}"
 
 demo = gr.Interface(
     fn=text_gen,
-        inputs=[url_input],
-    outputs=["text"])
-
-demo.launch(share=True)
+    inputs=[
+        gr.inputs.Textbox(label="Prompt", default="Why is the sky purple?"),
+        gr.inputs.Textbox(label="URL", default="http://198.175.88.52:8080/generate")
+    ],
+    outputs=gr.outputs.Textbox(label="Response"),
+    title="Text Generation",
+    description="Enter a prompt and a URL to generate text."
+)
 
+demo.launch()