Update app.py

app.py

@@ -1,24 +1,24 @@
 
 # from huggingface_hub import login, HfFolder
 import subprocess
-
+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}")
 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", "--no-deps", "trl peft accelerate bitsandbytes"], check=True)
+run_command(["pip", "install", "--no-deps", "trl", "peft", "accelerate", "bitsandbytes"]) 
 # 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[colab-new] @ git+https://github.com/unslothai/unsloth.git"], check=True)
-import subprocess
+# import subprocess
+
 
-# 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"])