Spaces:

hmb
/

zgpu-workflow

Runtime error

hmb HF Staff

Update app.py

445b558 verified 3 months ago

325 Bytes

import time

import gradio as gr
import spaces
import torch


@spaces.GPU(duration=10)
def generate(prompt: str) -> str:
    gpu = torch.cuda.get_device_name()
    time.sleep(5)
    return f"Generated on {gpu}: {prompt.upper()}"


demo = gr.Workflow(bind={"generate": generate})

if __name__ == "__main__":
    demo.launch()

	import time

	import gradio as gr
	import spaces
	import torch


	@spaces.GPU(duration=10)
	def generate(prompt: str) -> str:
	gpu = torch.cuda.get_device_name()
	time.sleep(5)
	return f"Generated on {gpu}: {prompt.upper()}"


	demo = gr.Workflow(bind={"generate": generate})

	if __name__ == "__main__":
	demo.launch()