add gpu

README.md

@@ -1,10 +1,10 @@
 ---
-title: Cyborgai Gradio
-emoji: 😻
-colorFrom: pink
-colorTo: indigo
+title: Zero Gpu Demo
+emoji: 🐢
+colorFrom: gray
+colorTo: green
 sdk: gradio
-sdk_version: 4.1.1
+sdk_version: 4.7.1
 app_file: app.py
 pinned: false
 license: apache-2.0

ai-comic-factory

@@ -1 +0,0 @@
-Subproject commit 11cdf4a219763f65eb38eb8f692a0f94eb973b43

app.py

@@ -5,8 +5,6 @@ import subprocess
 import torch
 import gradio as gr
 
-
-
 CUSTOM_CSS = """
 #output_box textarea {
     font-family: IBM Plex Mono, ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", monospace;
@@ -18,7 +16,7 @@ print(zero.device) # <-- 'cpu' 🤔
 
 @spaces.GPU
 def run_gpu() -> str:
-    print(zero.device) # <-- 'cuda:0' 
+    print(zero.device) # <-- 'cuda:0' 🤗
     output: str = ""
     try:
         output = subprocess.check_output(["nvidia-smi"], text=True)
@@ -36,19 +34,19 @@ def run(check: bool) -> str:
         comment = (
             datetime.datetime.now().replace(microsecond=0).isoformat().replace("T", " ")
         )
-        return f"# {comment}"
+        return f"# {comment}\n\nThis is running on CPU\n\nClick on 'Run on GPU' below to move to GPU instantly and run nvidia-smi"
 
 output = gr.Textbox(
     label="Command Output", max_lines=32, elem_id="output_box", value=run(False)
 )
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
-    #gr.Markdown("#### `zero-gpu`: how to run on serverless GPU for free on Spaces 🔥")
+    gr.Markdown("#### `zero-gpu`: how to run on serverless GPU for free on Spaces 🔥")
 
     output.render()
 
-    check = gr.Checkbox(label="Run")
+    check = gr.Checkbox(label="Run on GPU")
 
     check.change(run, inputs=[check], outputs=output, every=1)
 
-demo.queue().launch(show_api=True,share=True)
+demo.queue().launch(show_api=False)

requirements.txt

@@ -1,3 +0,0 @@
-torch 
-torchvision 
-torchaudio