Update app.py

app.py

@@ -7,6 +7,7 @@ import spaces
 peft_model_id = "rootxhacker/CodeAstra-7B"
 config = PeftConfig.from_pretrained(peft_model_id)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(device)
 model = AutoModelForCausalLM.from_pretrained(
     config.base_model_name_or_path,
     return_dict=True,