Spaces:
Runtime error
Runtime error
Manuel Calzolari
commited on
Commit
•
1ee69d0
1
Parent(s):
971627e
Fix app
Browse files
app.py
CHANGED
|
@@ -8,6 +8,8 @@ from langchain_core.prompts import PromptTemplate
|
|
| 8 |
from peft import PeftModel
|
| 9 |
from transformers import AutoModelForCausalLM, GenerationConfig, pipeline
|
| 10 |
|
|
|
|
|
|
|
| 11 |
# Define the embedding function
|
| 12 |
# I use the "all-MiniLM-L6-v2" model
|
| 13 |
embedding_function = SentenceTransformerEmbeddings(
|
|
|
|
| 8 |
from peft import PeftModel
|
| 9 |
from transformers import AutoModelForCausalLM, GenerationConfig, pipeline
|
| 10 |
|
| 11 |
+
base_model = "microsoft/phi-2"
|
| 12 |
+
|
| 13 |
# Define the embedding function
|
| 14 |
# I use the "all-MiniLM-L6-v2" model
|
| 15 |
embedding_function = SentenceTransformerEmbeddings(
|