Update app.py

app.py

@@ -197,7 +197,7 @@ def post_process_code(code):
 
 # Generate the final code from LaTeX
 def generate_code(latex_expression, max_length=512):
-    inputs = tokenizer(f"Latex Expression: {latex_expression} Solution:", return_tensors="pt").to("cuda")
+    inputs = tokenizer(f"Latex Expression: {latex_expression} Solution:", return_tensors="pt")
     outputs = model.generate(**inputs, max_length=max_length)
     generated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     post_processed_code = post_process_code(generated_code)