Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,8 @@ vectorstore.add_texts(texts=texts, metadatas=metadatas, embeddings=embeddings)
|
|
42 |
vectorstore.persist()
|
43 |
|
44 |
|
45 |
-
model_name = "deepseek-ai/deepseek-llm-7b-instruct"
|
|
|
46 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
47 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
|
48 |
|
|
|
42 |
vectorstore.persist()
|
43 |
|
44 |
|
45 |
+
# model_name = "deepseek-ai/deepseek-llm-7b-instruct"
|
46 |
+
model_name = "google/flan-t5-large"
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
48 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
|
49 |
|