saakshigupta commited on
Commit
1f955aa
·
verified ·
1 Parent(s): a9f2f98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ import torch
3
  import torch.nn as nn
4
  from torch.utils.data import DataLoader
5
  from torchvision import transforms
6
- from transformers.models.clip import CLIPModel # Updated import path
 
7
  from PIL import Image
8
  import numpy as np
9
  import io
 
3
  import torch.nn as nn
4
  from torch.utils.data import DataLoader
5
  from torchvision import transforms
6
+ from transformers import CLIPModel
7
+ from transformers.models.clip import CLIPModel
8
  from PIL import Image
9
  import numpy as np
10
  import io