Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import torch
 from utils import *
 from presets import *
-from transformers import Trainer, TrainingArguments, DataCollatorForLanguageModeling
+from transformers import Trainer, TrainingArguments, DataCollatorForLanguageModeling, BitsAndBytesConfig,
 from sentence_transformers.losses import CosineSimilarityLoss
 import numpy as np
 import evaluate