Update app.py

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
-import pandas as pd
-from datasets import load_dataset 
+
 import transformers
 from transformers import AutoProcessor
 from transformers import AutoModelForCausalLM
-import torch
-from torch.utils.data import Dataset, DataLoader
-from torchvision.transforms import Resize
+
 import os
 from PIL import Image
 saved_folder_path = "sudeep-007/saved_model"