Spaces:
Runtime error
Runtime error
Commit
•
b33888d
1
Parent(s):
7aaf734
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,6 @@ from torch import nn
|
|
| 8 |
from torch.utils.data import Dataset, DataLoader
|
| 9 |
from torchvision import transforms
|
| 10 |
from transformers import CLIPTextModel, CLIPTokenizer, CLIPVisionModel, CLIPModel
|
| 11 |
-
from transformers.pipelines import ImagePipeline
|
| 12 |
|
| 13 |
@spaces.GPU()
|
| 14 |
def train_image_generation_model(image_folder, text_folder, model_name="image_generation_model"):
|
|
|
|
| 8 |
from torch.utils.data import Dataset, DataLoader
|
| 9 |
from torchvision import transforms
|
| 10 |
from transformers import CLIPTextModel, CLIPTokenizer, CLIPVisionModel, CLIPModel
|
|
|
|
| 11 |
|
| 12 |
@spaces.GPU()
|
| 13 |
def train_image_generation_model(image_folder, text_folder, model_name="image_generation_model"):
|