File size: 1,812 Bytes
e598102 08bacf4 de57008 4a07312 08bacf4 e598102 72ecf70 096633b b2f3eb6 e598102 d26b817 de4a9bf d26b817 72ecf70 d26b817 72ecf70 e598102 72ecf70 08bacf4 35ba9e8 4af6d8d 35ba9e8 10b0e75 35ba9e8 10b0e75 9ca40b7 4a07312 10b0e75 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 |
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:
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")
time.sleep(86400) #Restart Space every 24 hours
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()
|