Update app.py

app.py

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