Update app.py

app.py

@@ -15,6 +15,8 @@ st.set_page_config(page_title="Repository Recommender", layout="wide")
 def load_model():
     model_name = "Salesforce/codet5-small"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
+    # Check if GPU is available
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     model = AutoModel.from_pretrained(model_name).to("cuda")
     return tokenizer, model