Spaces:
Running
Running
| import json | |
| import requests | |
| import gradio as gr | |
| import random | |
| import os | |
| import datetime | |
| from datetime import datetime | |
| API_TOKEN = os.getenv("API_TOKEN") | |
| headers = {"Authorization": f"Bearer {API_TOKEN}"} | |
| API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom" | |
| DECODEM_TOKEN=os.getenv("DECODEM_TOKEN") | |
| headers = {'Content-type': 'application/json', 'Accept': 'text/plain'} | |
| url_decodemprompts='https://us-central1-createinsightsproject.cloudfunctions.net/getdecodemprompts' | |
| data={"prompt_type":'future_in_words',"decodem_token":DECODEM_TOKEN} | |
| try: | |
| r = requests.post(url_decodemprompts, data=json.dumps(data), headers=headers) | |
| except requests.exceptions.ReadTimeout as e: | |
| print(e) | |
| #print(r.content) | |
| prompt=str(r.content, 'UTF-8') | |
| #print(prompt) | |
| def query(text): | |
| #global prompt | |
| print(text) | |
| print(datetime.today().strftime("%d-%m-%Y")) | |
| text = prompt+"\nInput:"+text + "\nOutput:" | |
| #print(text) | |
| payload = { | |
| "inputs": text, | |
| "parameters": { | |
| "max_length": 250, | |
| "temperature": 0.9, | |
| "seed":random.randint(0, 100), | |
| }, | |
| } | |
| data = json.dumps(payload) | |
| response = requests.request("POST", API_URL, headers=headers, data=data) | |
| #print('Response here\n',response.content) | |
| generated_text=json.loads(response.content.decode("utf-8"))[0]['generated_text'] | |
| result = generated_text.replace(text,'').strip() | |
| result = result.replace("Output:","") | |
| parts = result.split("###") | |
| topic = parts[0].strip() | |
| topic=topic.split("Input:")[0] | |
| print(topic) | |
| return topic | |
| with gr.Blocks() as demo: | |
| gr.Markdown("<h1><center>Ideas of the Future</center></h1>") | |
| gr.Markdown( | |
| """ChatGPT based Insights from <a href="https://www.decodem.ai">Decodem.ai</a> for businesses.\nWhile ChatGPT has multiple use cases we have evolved specific use cases/ templates for businesses \n\n This template provides ideas on how a business would look like in the future. Enter a business area and get the results. We use a equally powerful AI model bigscience/bloom""" | |
| ) | |
| textbox = gr.Textbox(placeholder="Enter business type here...", lines=1,label='Your business area') | |
| btn = gr.Button("Generate") | |
| output1 = gr.Textbox(lines=2,label='The future') | |
| btn.click(query,inputs=[textbox], outputs=[output1]) | |
| examples = gr.Examples(examples=['icecream parlor','space travel','book shop','ecommerce','grocery delivery'], | |
| inputs=[textbox]) | |
| demo.launch() |