micole66 commited on
Commit
eb7c784
1 Parent(s): acc6696

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import gradio as gr
2
  from transformers import AutoProcessor, AutoModel
3
 
4
- processor = AutoProcessor.from_pretrained("flax-community/clip-rsicd-v2")
5
 
6
- model = AutoModel.from_pretrained("flax-community/clip-rsicd-v2")
7
 
8
 
9
  def calculate_score(image, text):
 
1
  import gradio as gr
2
  from transformers import AutoProcessor, AutoModel
3
 
4
+ processor = AutoProcessor.from_pretrained("laion/CLIP-ViT-H-14-laion2B-s32B-b79K")
5
 
6
+ model = AutoModel.from_pretrained("laion/CLIP-ViT-H-14-laion2B-s32B-b79K")
7
 
8
 
9
  def calculate_score(image, text):