update

app.py

@@ -40,7 +40,7 @@ def generate(dna_sequence):
     dna_sequence = dna_sequence + "[MASK]"
     tokenized_message = tokenizer(
         [dna_sequence], return_tensors='pt', return_token_type_ids=False, add_special_tokens=True
-    ).to('cuda')
+    ).to(device)
 
     response = model.generate(**tokenized_message, max_new_tokens=110, do_sample=False)
     reply = tokenizer.batch_decode(response, skip_special_tokens=True)[0]