RobotJelly commited on
Commit
06ffe20
1 Parent(s): 37a1406
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from PIL import Image
7
  from io import BytesIO
8
  import requests
9
  import gradio as gr
 
10
  from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
11
 
12
  # check if CUDA available
7
  from io import BytesIO
8
  import requests
9
  import gradio as gr
10
+ import os
11
  from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
12
 
13
  # check if CUDA available