File size: 1,828 Bytes
e598102 08bacf4 de57008 4a07312 08bacf4 e598102 72ecf70 096633b b2f3eb6 e598102 d26b817 de4a9bf d26b817 72ecf70 d26b817 72ecf70 e598102 72ecf70 08bacf4 35ba9e8 69165c2 35ba9e8 e09a781 35ba9e8 10b0e75 4ac0888 10b0e75 35ba9e8 10b0e75 9ca40b7 4a07312 4ac0888 72ecf70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from gradio_client import Client
import time
import os
import gradio as gr
from threading import Thread
print("RAN")
hf_token_id = os.environ['TOKEN']
client = Client("https://reverent-atlei2.hf.space/",hf_token=hf_token_id)
def warmer():
while True:
try:
text_output,image_output = client.predict(
"atleiwarm0001", # str in 'message' Textbox component
"", # str (filepath or URL to image) in 'Upload any Image' Image component
api_name="/predict"
)
text_output_2,image_output_2 = client.predict(
"atleiwarm0002", # str in 'message' Textbox component
"", # str (filepath or URL to image) in 'Upload any Image' Image component
api_name="/predict"
)
if (text_output != None) and (text_output_2 != None):
print("ALIVE")
print("pygmalion response : ", text_output)
print("qna response : ", text_output_2)
else:
print("DISCONNECTED")
time.sleep(300)
except Exception as e:
print("NO INTERNET")
print(str(e))
def restart():
while True:
time.sleep(10800) # Restart ATLEI2 Space every 3 hours
try:
text_output,image_output = client.predict(
"atleiwarm0003", # str in 'message' Textbox component
"", # str (filepath or URL to image) in 'Upload any Image' Image component
api_name="/predict"
)
print("RESTARTED")
except Exception as e:
print("NO INTERNET")
print(str(e))
def placeholder(dummy):
return "lol"
t = Thread(target=warmer)
r = Thread(target=restart)
t.start()
r.start()
iface = gr.Interface(fn=placeholder, inputs="text", outputs="text")
iface.launch()
|