File size: 1,951 Bytes
e598102 08bacf4 de57008 4a07312 08bacf4 e598102 72ecf70 096633b b2f3eb6 e598102 d26b817 de4a9bf d26b817 72ecf70 d26b817 72ecf70 e598102 72ecf70 08bacf4 e09a781 35ba9e8 e09a781 35ba9e8 e09a781 35ba9e8 e09a781 35ba9e8 10b0e75 e09a781 10b0e75 35ba9e8 10b0e75 9ca40b7 4a07312 e09a781 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 61 62 63 64 65 66 |
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))
counter = 0
def restart():
global counter
while True:
counter = 0
while counter <= 86400: #Restart Space every 24 hours
counter += 1
time.sleep(1)
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):
global counter
return counter
t = Thread(target=warmer)
r = Thread(target=restart)
t.start()
r.start()
iface = gr.Interface(fn=placeholder, inputs="text", outputs=counter)
iface.launch()
|