Update app.py

app.py

@@ -1,9 +1,16 @@
 #ref: https://huggingface.co/blog/AmelieSchreiber/esmbind
 import gradio as gr
 
+from transformers import Trainer, TrainingArguments, AutoTokenizer, EsmForMaskedLM, TrainerCallback
+from torch.utils.data import DataLoader, Dataset, RandomSampler
+import pandas as pd
+import torch
+from torch.optim import AdamW
+import wandb
+import numpy as np
+
+'''
 import os
-# os.environ["CUDA_VISIBLE_DEVICES"] = "0"
-#import wandb
 import numpy as np
 import torch
 import torch.nn as nn
@@ -34,6 +41,7 @@ from accelerate import Accelerator
 from peft import get_peft_config, PeftModel, PeftConfig, get_peft_model, LoraConfig, TaskType
 
 from plot_pdb import plot_struc
+'''
 
 def suggest(option):
     if option == "Plastic degradation protein":