Spaces:

jslin09
/

legal_document_drafting

Running on Zero

jslin09 commited on Mar 16, 2023

Commit

d97db75

1 Parent(s): bccb339

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,11 @@
 import gradio as gr
-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"]

 import gradio as gr
+from transformers import pipeline, set_seed
 generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-fraud')
 def generate(text):
+    seed = random.randint(10, 500)
+    set_seed(seed)
     result = generator(text, max_length=500, num_return_sequences=1)
     return result[0]["generated_text"]