ejschwartz commited on
Commit
81d30d3
·
1 Parent(s): 02ecf35
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import transformers
4
  pipe = transformers.pipeline("text2text-generation", model="ejschwartz/slade-x86-O3")
5
 
6
  def predict(asm):
7
- return pipe(asm, max_length=1024)
 
8
 
9
  demo = gr.Interface(fn=predict, inputs="text", outputs="text")
10
  demo.queue()
 
4
  pipe = transformers.pipeline("text2text-generation", model="ejschwartz/slade-x86-O3")
5
 
6
  def predict(asm):
7
+ o = pipe(asm, max_length=1024)
8
+ return o[0]['generated_text']
9
 
10
  demo = gr.Interface(fn=predict, inputs="text", outputs="text")
11
  demo.queue()