Spaces:
Runtime error
Runtime error
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 |
|