Fixes

.gitignore

@@ -1 +1,2 @@
 .env/
+dev.env

.vscode/settings.json

@@ -2,5 +2,6 @@
   "[python]": {
     "editor.defaultFormatter": "ms-python.black-formatter"
   },
-  "editor.formatOnSave": true
+  "editor.formatOnSave": true,
+  "python.envFile": "${workspaceFolder}/dev.env"
 }

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 4.37.1
+sdk_version: 4.43.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,5 +1,6 @@
 import time
 import os
+from typing import Literal, Tuple
 import gradio as gr
 import torch
 from transformers import AutoModel, AutoTokenizer
@@ -22,7 +23,9 @@ meilisearch_index = meilisearch_client.index(meilisearch_index_name)
 output_options = ["RAG-friendly", "human-friendly"]
 
 
-def search_embeddings(query_text, output_option):
+def search_embeddings(
+    query_text: str, output_option: Literal["RAG-friendly", "human-friendly"]
+) -> Tuple[str, str]:
     start_time_embedding = time.time()
     query_prefix = "Represent this sentence for searching code documentation: "
     query_tokens = tokenizer(
@@ -99,7 +102,7 @@ demo = gr.Interface(
         ),
     ],
     outputs=[gr.Markdown(), gr.Markdown()],
-    title="HF Docs Emebddings Explorer",
+    title="HF Docs Embeddings Explorer",
     allow_flagging="never",
 )