Pranjal Gupta commited on
Commit
b277fbe
·
1 Parent(s): d272c6c

using gpt2- as it was not gated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "meta-llama/Llama-3.2-1B-Instruct"
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(