Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py
+# app.py (corrected version)
 
 # Handle missing dependencies first
 try:
@@ -13,7 +13,10 @@ try:
     import bitsandbytes
 except ImportError as e:
     missing_package = str(e).split("'")[-2]  # Extract the missing package name
-    os.system(f'pip install "{missing_package}>=0.26.0" if "accelerate" in missing_package else f'pip install {missing_package}')
+    if "accelerate" in missing_package:
+        os.system(f'pip install "accelerate>=0.26.0"')
+    else:
+        os.system(f'pip install "{missing_package}"')
     # Re-import after installation
     import gradio as gr
     from transformers import LlamaForCausalLM, LlamaTokenizer, Trainer, TrainingArguments
@@ -25,7 +28,8 @@ except ImportError as e:
     from accelerate import Accelerator
     import bitsandbytes
 
-# Model setup
+# Rest of your script remains the same...
+# Model setup, training function, Gradio UI, etc., as shown in your previous script
 MODEL_ID = "meta-llama/Llama-2-7b-hf"  # Use Llama-2-7b; switch to "meta-llama/Llama-3-8b-hf" for Llama 3
 tokenizer = LlamaTokenizer.from_pretrained(MODEL_ID)