sudeep-007 commited on
Commit
b9bb3e9
·
1 Parent(s): 4a51cdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,7 +3,9 @@ import gradio as gr
3
  import transformers
4
  from transformers import AutoProcessor
5
  from transformers import AutoModelForCausalLM
6
-
 
 
7
  import os
8
  from PIL import Image
9
  saved_folder_path = "sudeep-007/saved_model"
 
3
  import transformers
4
  from transformers import AutoProcessor
5
  from transformers import AutoModelForCausalLM
6
+ import torch
7
+ from torch.utils.data import Dataset, DataLoader
8
+ from torchvision.transforms import Resize
9
  import os
10
  from PIL import Image
11
  saved_folder_path = "sudeep-007/saved_model"