Spaces:

cbensimon
/

zero-gpu-test0

Runtime error

Update app.py

b0f6b93 about 3 years ago

218 Bytes

import gradio as gr
import spaces
import torch

@spaces.GPU
def greet(name):
    return f"Tensor: {torch.Tensor([0]).cuda()}, Name: {name}"

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

	import gradio as gr
	import spaces
	import torch

	@spaces.GPU
	def greet(name):
	return f"Tensor: {torch.Tensor([0]).cuda()}, Name: {name}"

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