Fix setting model device for CUDA

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 
-model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona")
-tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
+model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
+tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
+
 LANG_CODES = {
     "English":"en",
     "toki pona":"tl"