sohojoe commited on
Commit
623a026
1 Parent(s): 106b424

re add some requirments

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +3 -3
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
  import torch
3
  from PIL import Image
4
- from torchvision import transforms
5
  # from diffusers import StableDiffusionPipeline, StableDiffusionImageVariationPipeline, DiffusionPipeline
6
  import numpy as np
7
  import pandas as pd
8
  import math
9
- from transformers import CLIPTextModel, CLIPTokenizer
10
  import os
11
 
12
  from clip_retrieval.clip_client import ClipClient, Modality
 
1
  import gradio as gr
2
  import torch
3
  from PIL import Image
4
+ # from torchvision import transforms
5
  # from diffusers import StableDiffusionPipeline, StableDiffusionImageVariationPipeline, DiffusionPipeline
6
  import numpy as np
7
  import pandas as pd
8
  import math
9
+ # from transformers import CLIPTextModel, CLIPTokenizer
10
  import os
11
 
12
  from clip_retrieval.clip_client import ClipClient, Modality
requirements.txt CHANGED
@@ -1,10 +1,10 @@
1
  # --extra-index-url https://download.pytorch.org/whl/cu113
2
- # torch
3
  # # torchvision
4
- # numpy
5
  # transformers
6
  # # diffusers
7
  # # ftfy
8
  # gradio
9
  # accelerate
10
- # clip-retrieval == 2.36.1
 
1
  # --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch
3
  # # torchvision
4
+ numpy
5
  # transformers
6
  # # diffusers
7
  # # ftfy
8
  # gradio
9
  # accelerate
10
+ clip-retrieval == 2.36.1