Spaces:
Runtime error
Runtime error
Catherine ZHOU
commited on
Commit
·
d74fc9d
1
Parent(s):
5adbc30
update requirements
Browse files- app.py +2 -2
- requirements.txt +10 -1
app.py
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
from gradio.flagging import FlaggingCallback
|
|
|
|
|
3 |
from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
|
4 |
from sentence_transformers import util
|
5 |
import pickle
|
@@ -8,10 +10,8 @@ import os
|
|
8 |
import logging
|
9 |
import csv
|
10 |
import datetime
|
11 |
-
from gradio_client import utils as client_utils
|
12 |
from pathlib import Path
|
13 |
from typing import List, Any
|
14 |
-
from gradio.components import IOComponent
|
15 |
|
16 |
class SaveRelevanceCallback(FlaggingCallback):
|
17 |
""" Callback to save the image relevance state to a csv file
|
|
|
1 |
import gradio as gr
|
2 |
from gradio.flagging import FlaggingCallback
|
3 |
+
from gradio.components import IOComponent
|
4 |
+
from gradio_client import utils as client_utils
|
5 |
from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
|
6 |
from sentence_transformers import util
|
7 |
import pickle
|
|
|
10 |
import logging
|
11 |
import csv
|
12 |
import datetime
|
|
|
13 |
from pathlib import Path
|
14 |
from typing import List, Any
|
|
|
15 |
|
16 |
class SaveRelevanceCallback(FlaggingCallback):
|
17 |
""" Callback to save the image relevance state to a csv file
|
requirements.txt
CHANGED
@@ -1,4 +1,13 @@
|
|
1 |
transformers
|
2 |
sentence-transformers
|
3 |
torch
|
4 |
-
gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
transformers
|
2 |
sentence-transformers
|
3 |
torch
|
4 |
+
gradio
|
5 |
+
gradio_client
|
6 |
+
pickle
|
7 |
+
PIL
|
8 |
+
os
|
9 |
+
logging
|
10 |
+
csv
|
11 |
+
datetime
|
12 |
+
pathlib
|
13 |
+
typing
|