Update app.py

app.py

@@ -1,5 +1,6 @@
 import time
 last_run = 0
+olast_run = 0
 import os
 from huggingface_hub import HfApi
 api = HfApi(
@@ -15,8 +16,19 @@ def reboot():
         return "Rebooted"
     else:
         raise gr.Error("The Space has already been rebooted recently")
+def rebootorig():
+    global olast_run
+    current_time = time.time()
+    if current_time - olast_run >= 300:
+        api.restart_space('mrfakename/OpenDalle-GPU-Demo', factory_reboot=True)
+        olast_run = current_time
+        return "Rebooted"
+    else:
+        raise gr.Error("The Space has already been rebooted recently")
 with gr.Blocks() as demo:
     gr.Markdown("# Request a Reboot of OpenDalle v1.1 GPU Demo\n\nIs [the demo](https://huggingface.co/spaces/mrfakename/OpenDalleV1.1-GPU-Demo) not working? Request a reboot!")
-    btn = gr.Button("Reboot")
+    btn = gr.Button("Reboot V1.1 GPU DEMO")
     btn.click(reboot, outputs=[btn])
+    btnorig = gr.Button("Reboot V1 (original) GPU DEMO")
+    btnorig.click(rebootorig, outputs=[btnorig])
 demo.launch()