Spaces:

jslin09
/

legal_document_drafting

Running

jslin09 commited on Mar 15, 2023

Commit

c6c3f60

•

1 Parent(s): 44f8057

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,8 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-fraud')
 def generate(text):
     result = generator(text, max_length=500, num_return_sequences=1)
     return result[0]["generated_text"]

 generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-fraud')
+upstream = gr.Interface.load("jslin09/bloom-560m-finetuned-fraud")
 def generate(text):
     result = generator(text, max_length=500, num_return_sequences=1)
     return result[0]["generated_text"]