ai-api-test / main.py
Rupnil's picture
Update main.py
b476618
raw
history blame
688 Bytes
from flask import Flask
from gradio_client import Client
import time
from multiprocessing import Process
app = Flask(__name__)
@app.route('/')
def hello():
return '{"status": {"details": {"code": 200, "message": "working"}}}'
@app.route('/api/<string:query>')
def api(query):
client = Client("https://theblueberry-ai-mist-chat.hf.space/")
result = client.predict(
query, # str in 'Message' Textbox component
api_name = "/chat"
)
return result
def record_loop():
while True:
print("loop running")
time.sleep(5)
if __name__ == "__main__":
p = Process(target=record_loop)
p.start()
app.run(debug=True, use_reloader=False)
p.join()