Update app.py

app.py

@@ -8,8 +8,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("AhmedSSoliman/MarianCG_NL2Code")
 
 def generate_code(NL):
     inputs = tokenizer([NL], padding="max_length", truncation=True, max_length=512, return_tensors="pt")
-    input_ids = inputs.input_ids.to("cuda")
-    attention_mask = inputs.attention_mask.to("cuda")
+    input_ids = inputs.input_ids
+    attention_mask = inputs.attention_mask
     outputs = new_model.generate(input_ids, attention_mask=attention_mask)
     # all special tokens including will be removed
     output_code = tokenizer.batch_decode(outputs, skip_special_tokens=True)