Spaces:
Sleeping
Sleeping
Pranjal Gupta
commited on
Commit
·
b277fbe
1
Parent(s):
d272c6c
using gpt2- as it was not gated
Browse files
app.py
CHANGED
|
@@ -87,7 +87,7 @@ def using_ollama_model(retriever, query, results, conversation_history):
|
|
| 87 |
)
|
| 88 |
|
| 89 |
doc_texts = "\\n".join([doc.page_content for doc in results])
|
| 90 |
-
model_id = "
|
| 91 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 92 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 93 |
pipe = pipeline(
|
|
|
|
| 87 |
)
|
| 88 |
|
| 89 |
doc_texts = "\\n".join([doc.page_content for doc in results])
|
| 90 |
+
model_id = "gpt2"
|
| 91 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 92 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 93 |
pipe = pipeline(
|