Update app.py

app.py

@@ -46,8 +46,6 @@ i_weights = params["i_weights"]
 i_length = params["i_length"]
 # Load main model
 model = load_models()
-model.model.to("cuda")
-model.device = "cuda"
 tokenizer = Tokenizer(f'{weight_path}/tokenizer.model')
 # Create ngram models
 ngrams = make_models("ckpts-200k", bigram=True, trigram=True, fourgram=True, fivegram=True, sixgram=True, sevengram=False)