File size: 1,306 Bytes
e598102
08bacf4
 
de57008
4a07312
08bacf4
 
 
 
e598102
72ecf70
 
 
096633b
b2f3eb6
e598102
 
 
d26b817
 
 
 
 
 
72ecf70
d26b817
 
72ecf70
 
e598102
72ecf70
 
 
08bacf4
10b0e75
 
 
 
 
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
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(
				"atleiwarm0001",	# 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 placeholder(dummy):
    return "lol"
    
t = Thread(target=warmer)
t.start()

iface = gr.Interface(fn=placeholder, inputs="text", outputs="text")
iface.launch()