Update app.py

app.py

@@ -6,7 +6,7 @@ import random
 
 model_name = "bloom-560m"
 model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
-tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>')
+tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
 
 def rnd_generate(prompt):
     rnd_seed = random.randint(10, 500)
@@ -65,5 +65,4 @@ with gr.Blocks() as demo:
             btn.click(rnd_generate, inputs=[result2], outputs=[result2])
     
 if __name__ == "__main__":
-    demo.launch()
-#    gr.Interface.load("models/jslin09/bloom-560m-finetuned-fraud").launch()
+    demo.launch()