Update app.py

app.py

@@ -17,11 +17,6 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 BASETEN_API = os.environ.get("BASETEN_API", None)
 BASETEN_KEY = os.environ.get("BASETEN_KEY", None)
 
-# Set the device to GPU or CPU
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-
-
 #st.set_page_config(layout="wide")
 # Load custom CSS to integrate Bootstrap, Font Awesome, and Google Fonts
 st.markdown('''