updated changes

app.py

@@ -1,13 +1,14 @@
 from peft import PeftModel, PeftConfig
+import torch
+from datasets import load_dataset
 
 torch.cuda.is_available()
 print("executed successfully")
 
-from datasets import load_dataset
 dataset_name = "timdettmers/openassistant-guanaco"
 dataset = load_dataset(dataset_name, split="train")
 
-import torch
+
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
 # quantizition configuration
@@ -36,4 +37,3 @@ inputs = tokenizer(text, return_tensors="pt").to(device)
 outputs = model.generate(**inputs, max_new_tokens=50)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
-