Update app.py

app.py

@@ -10,8 +10,8 @@ import random
 #model_name = "bloom-560m"
 #model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud').to('cuda')
 #tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
-tokenizer = AutoTokenizer.from_pretrained("jslin09/gemma2-2b-fraud").to('cuda')
-model = AutoModelForCausalLM.from_pretrained("jslin09/gemma2-2b-fraud")
+tokenizer = AutoTokenizer.from_pretrained("jslin09/gemma2-2b-fraud")
+model = AutoModelForCausalLM.from_pretrained("jslin09/gemma2-2b-fraud").to('cuda')
 
 @spaces.GPU
 def rnd_generate(prompt):