Update app.py

app.py

@@ -14,8 +14,8 @@ import requests
 
 from sentence_transformers import SentenceTransformer, util
 
-#model_sts = SentenceTransformer('stsb-distilbert-base')
-model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
+model_sts = SentenceTransformer('stsb-distilbert-base')
+# model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
 #batch_size = 1
 #scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)
 
@@ -136,8 +136,8 @@ def Visual_re_ranker(caption_man, caption_woman, context_label, context_prob):
 demo = gr.Interface(
     fn=Visual_re_ranker,
     description="Demo for Women Wearing Lipstick: Measuring the Bias Between Object and Its Related Gender",
-    # inputs=[gr.Textbox(value="a man riding a motorcycle on a road") , gr.Textbox(value="a woman riding a motorcycle on a road"), gr.Textbox(value="motor scooter"),  gr.Textbox(value="0.2183")],
-    inputs=[gr.Textbox(value="a man playing a video game in a living room") , gr.Textbox(value=" a woman playing a video game in a living room"), gr.Textbox(value="joystick"),  gr.Textbox(value="0.2732")],
+    inputs=[gr.Textbox(value="a man riding a motorcycle on a road") , gr.Textbox(value="a woman riding a motorcycle on a road"), gr.Textbox(value="motor scooter"),  gr.Textbox(value="0.2183")],
+  
   
    
     outputs="label",