Catherine ZHOU commited on
Commit
d74fc9d
·
1 Parent(s): 5adbc30

update requirements

Browse files
Files changed (2) hide show
  1. app.py +2 -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