adding scheduler

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import torch
-from utils import load_model, generate_random_img
+from utils import load_model, generate_random_img, schedule_function
+import time
+import random
+import threading 
+from gradio_client import Client
 
 def generate_image():
     with torch.no_grad():
@@ -18,5 +22,7 @@ iface = gr.Interface(
     css='img_styles.css',
 )
 
-iface.launch()
-
+if __name__ == '__main__':
+    scheduler_thread = threading.Thread(target=schedule_function) # avoiding sleep, again this project is for academic purposes only
+    scheduler_thread.start()
+    iface.launch()

utils.py

@@ -1,5 +1,7 @@
-# Generator model
-# Critic Model
+import time
+import random
+import threading 
+from gradio_client import Client
 
 import torch
 import torchvision
@@ -142,6 +144,19 @@ def generate_random_img(model):
 
   return generated_image
 
+def schedule_function(): # for dummy space so that this spcae and other space will call each other to avoid sleep time, again this project is for academic purpose.
+  while True:
+    # wait_time = random.uniform(3 * 60 * 60, 5 * 60 * 60)  # Get a random wait time between 3 and 5 hours in seconds
+    wait_time = random.uniform(3, 5)
+    time.sleep(wait_time)
+    # call dummyscape
+    client = Client("https://huseyng-dummyspace.hf.space/")
+    result = client.predict(
+                    f"Howdy! {wait_time}",	# str representing string value in 'name' Textbox component
+                    api_name="/predict"
+    )
+    print(result)
+
 if __name__ == "__main__":
     model = load_model('generator','generator_model_epoch_94.pth')
     generate_random_img(model)