dipeshpratapsingh's picture
Update app.py
d52cb50 verified
raw
history blame contribute delete
780 Bytes
import gradio as gr
import requests
import uuid
session_id = str(uuid.uuid4())
def generate_text(prompt):
url = "https://mgpzj8t1f2.execute-api.us-east-1.amazonaws.com/DEV/generate"
headers = {"Content-Type": "application/json"}
data = {"prompt": prompt, "sessionId": session_id}
response = requests.post(url, headers=headers, json=data)
response.raise_for_status() # Ensure we raise an error for bad responses
return response.text.replace('"', '').replace("\\n", ' ').replace('/',' ').replace('\\', ' ')
iface = gr.Interface(
fn=generate_text,
inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
outputs=gr.Textbox(label="Generated Text"),
live=False,
allow_flagging="never"
)
iface.launch()