add rerank test

README.md

@@ -11,3 +11,11 @@ license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+# usage
+```
+git add .
+git commit -m "comment"
+git push
+```

app.py

@@ -1,4 +1,30 @@
-import streamlit as st
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+# 加载rerank模型和tokenizer
+model_name = "BAAI/bge-reranker-v2-m3"  # 替换为你的rerank模型名称
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+# 定义候选项和查询
+query = "What is the capital of France?"
+candidates = [
+    "Paris is the capital of France.",
+    "Berlin is the capital of Germany.",
+    "Madrid is the capital of Spain."
+]
+
+# 对每个候选项进行打分
+scores = []
+for candidate in candidates:
+    inputs = tokenizer(query, candidate, return_tensors="pt", truncation=True)
+    with torch.no_grad():
+        logits = model(**inputs).logits
+    scores.append(logits.item())
+
+# 根据分数对候选项重新排序
+ranked_candidates = [x for _, x in sorted(zip(scores, candidates), reverse=True)]
+
+# 输出排序结果
+for i, candidate in enumerate(ranked_candidates):
+    print(f"Rank {i + 1}: {candidate} (Score: {scores[i]})")

demo.py

@@ -0,0 +1,40 @@
+import streamlit as st
+import base64
+
+from streamlit.components.v1 import html
+
+st.title("Video Player App")
+
+video_file_path = "D:\\huggingface\\KHome\\digitalhuman.mp4"
+# 读取视频文件的内容
+with open(video_file_path, "rb") as video_file:
+    video_bytes = video_file.read()
+    video_base64 = base64.b64encode(video_bytes).decode('utf-8')
+
+STOPPED_STATE = f"""
+    <video id="videoPlayer" width="320" height="240" controls>
+      <source src="data:video/mp4;base64,{video_base64}" type="video/mp4">
+      Your browser does not support the video tag.
+    </video>
+    """
+
+PLAYING_STATE = f"""
+    <video id="videoPlayer" width="320" height="240" controls autoplay>
+      <source src="data:video/mp4;base64,{video_base64}" type="video/mp4">
+      Your browser does not support the video tag.
+    </video>
+    """
+
+
+playstate = st.session_state.get("playstate")
+if not playstate:
+    playstate = STOPPED_STATE
+st.markdown(playstate, unsafe_allow_html=True)
+
+# 播放按钮
+if st.button("Play"):
+    st.session_state["playstate"] = PLAYING_STATE
+
+# 暂停按钮
+if st.button("Pause"):
+    st.session_state["playstate"] = STOPPED_STATE

stablediffusion3.py

@@ -0,0 +1,3 @@
+from diffusers import DiffusionPipeline
+
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium")