Update app.py

app.py

@@ -5,7 +5,7 @@ from datasets import load_dataset
 from huggingface_hub import HfApi, login
 #from peft import AutoPeftModelForCausalLM, LoraConfig
 #from random import randint
-#from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, TrainingArguments, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM#, BitsAndBytesConfig, TrainingArguments, pipeline
 #from trl import SFTTrainer, setup_chat_format
 
 hf_profile = "bstraehle"