Oysiyl commited on
Commit
7bf3496
1 Parent(s): e81c4be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,7 +1,16 @@
1
  import gradio as gr
2
 
 
 
 
 
 
 
3
  def greet(name):
4
- return "Hello " + name + "!!"
 
 
 
5
 
6
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
7
  demo.launch()
 
1
  import gradio as gr
2
 
3
+ from transformers import AutoTokenizer
4
+ from transformers import AutoModelForSeq2SeqLM
5
+
6
+ tokenizer = AutoTokenizer.from_pretrained("elvish-translator-quenya-t5-small")
7
+ model = AutoModelForSeq2SeqLM.from_pretrained("elvish-translator-quenya-t5-small")
8
+
9
  def greet(name):
10
+ inputs = tokenizer(text, return_tensors="pt").input_ids
11
+ outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
12
+ result = tokenizer.decode(outputs[0], skip_special_tokens=True)
13
+ return result
14
 
15
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
16
  demo.launch()