feng2022 commited on
Commit
30bd241
·
1 Parent(s): a30d3cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from datasets import Dataset
3
  import gradio as gr
4
  from pynvml import *
5
 
 
 
6
 
7
  def print_gpu_utilization():
8
  nvmlInit()
 
3
  import gradio as gr
4
  from pynvml import *
5
 
6
+ def predict(text):
7
+ return pipe(text)[0]["translation_text"]
8
 
9
  def print_gpu_utilization():
10
  nvmlInit()