Vivien commited on
Commit
b59b1d0
1 Parent(s): 3705c76

Simplify code

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +1 -2
.gitignore ADDED
@@ -0,0 +1 @@
 
1
+ .vscode/
app.py CHANGED
@@ -2,11 +2,10 @@ import streamlit as st
2
  import pandas as pd, numpy as np
3
  from html import escape
4
  import os
5
- from transformers import CLIPProcessor, CLIPTextModel, CLIPModel
6
 
7
  @st.cache(show_spinner=False,
8
  hash_funcs={CLIPModel: lambda _: None,
9
- CLIPTextModel: lambda _: None,
10
  CLIPProcessor: lambda _: None,
11
  dict: lambda _: None})
12
  def load():
2
  import pandas as pd, numpy as np
3
  from html import escape
4
  import os
5
+ from transformers import CLIPProcessor, CLIPModel
6
 
7
  @st.cache(show_spinner=False,
8
  hash_funcs={CLIPModel: lambda _: None,
 
9
  CLIPProcessor: lambda _: None,
10
  dict: lambda _: None})
11
  def load():