Update app.py

app.py

@@ -1,15 +1,39 @@
 
-# from huggingface_hub import login, HfFolder
+# # from huggingface_hub import login, HfFolder
+# import subprocess
+
+# subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "bitsandbytes"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
+# subprocess.run(["pip", "install", "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git"], check=True)
 import subprocess
 
-subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "bitsandbytes"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
-subprocess.run(["pip", "install", "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git"], check=True)
+def run_command(cmd, shell=False):
+    """Runs a shell command and prints the output."""
+    try:
+        result = subprocess.run(cmd, shell=shell, capture_output=True, text=True, check=True)
+        print(result.stdout)
+    except subprocess.CalledProcessError as e:
+        print(f"Error executing command: {e.stderr}")
+
+# 1. Create the conda environment
+run_command("conda create -y --name unsloth_env python=3.10")
+
+# 2. Activate the environment (Note: Requires shell=True)
+run_command("conda activate unsloth_env", shell=True)  
+
+# 3. Install PyTorch and related packages with conda
+run_command("conda install pytorch-cuda=<12.1/11.8> pytorch cudatoolkit xformers -c pytorch -c nvidia -c xformers")
+
+# 4. Install unsloth from the GitHub repository with pip
+run_command("pip install \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"")
+
+# 5. Install additional pip packages without dependencies
+run_command("pip install --no-deps trl peft accelerate bitsandbytes")
 
 import os
 HF_TOKEN = os.environ["HF_TOKEN"]