re add some requirments
Browse files- app.py +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 |
-
|
3 |
# # torchvision
|
4 |
-
|
5 |
# transformers
|
6 |
# # diffusers
|
7 |
# # ftfy
|
8 |
# gradio
|
9 |
# accelerate
|
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
|