Spaces:

universalsoftware
/

uchat

Running

shoom013 commited on Sep 16, 2024

Commit

38bc394

verified ·

1 Parent(s): 17fedb2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ Settings.llm = HuggingFaceLLM(model_name="stabilityai/stablelm-zephyr-3b",
                               tokenizer_name="stabilityai/stablelm-zephyr-3b",
                              )
-Settings.embed_model = HuggingFaceEmbedding(model_name="facebook/blenderbot-400M-distill")
 documents = [Document(text="Indian parliament elections happened in April-May 2024. BJP Party won."),
              Document(text="Indian parliament elections happened in April-May 2021. XYZ Party won."),
              Document(text="Indian parliament elections happened in 2020. ABC Party won."),

                               tokenizer_name="stabilityai/stablelm-zephyr-3b",
                              )
+Settings.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
 documents = [Document(text="Indian parliament elections happened in April-May 2024. BJP Party won."),
              Document(text="Indian parliament elections happened in April-May 2021. XYZ Party won."),
              Document(text="Indian parliament elections happened in 2020. ABC Party won."),