Update app.py

app.py

@@ -3,7 +3,7 @@
 
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, HfArgumentParser,TrainingArguments,pipeline, logging, TextStreamer, MistralForCausalLM
 from peft import LoraConfig, PeftModel, prepare_model_for_kbit_training, get_peft_model,AutoPeftModelForCausalLM
-import os,torch, wandb, platform, gradio, warnings
+import os,torch, platform, warnings
 from datasets import load_dataset
 from trl import SFTTrainer
 from huggingface_hub import notebook_login