Update app.py

app.py

@@ -9,7 +9,7 @@ token = os.environ['TOKEN']
 MODEL_NAME = "atlasia/Al-Atlas-LLM"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=token)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=token)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=token).to('cuda')
 
 # Predefined examples
 examples = [