Spaces:

cbensimon
/

pytorch-without-cuda

Runtime error

pytorch-without-cuda / app.py

cbensimon HF staff

Update app.py

9ab1399 over 1 year ago

No virus

307 Bytes

	import gradio as gr
	import torch
	from transformers import pipeline

	print(f'{torch.cuda.is_available()=}')

	pipe = pipeline('translation_en_to_fr', device=0)

	def translate(text):
	return pipe(text)[0]['translation_text']

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