Sagar Sanghani commited on
Commit
52cb9d6
·
1 Parent(s): b78b0a3

roll back model to qwen

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. model.py +2 -2
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__/*
model.py CHANGED
@@ -107,7 +107,7 @@ class Model:
107
  ]
108
 
109
  llm = HuggingFaceEndpoint(
110
- repo_id="meta-llama/Llama-2-7b-chat-hf",
111
  huggingfacehub_api_token=self.token,
112
  temperature=0
113
  )
@@ -134,7 +134,7 @@ class Model:
134
  def main():
135
  #load_dotenv(find_dotenv())
136
  model = Model()
137
- response = model.get_answer("In what directory could the public find the free Penguin Project dating software for Mac as of 1997?")
138
  print(f"the output is: {response}")
139
 
140
  if __name__ == "__main__":
 
107
  ]
108
 
109
  llm = HuggingFaceEndpoint(
110
+ repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
111
  huggingfacehub_api_token=self.token,
112
  temperature=0
113
  )
 
134
  def main():
135
  #load_dotenv(find_dotenv())
136
  model = Model()
137
+ response = model.get_answer("How many albums were published by mercedes sosa between 2000 and 2009 you can use wikipedia to figure it out?")
138
  print(f"the output is: {response}")
139
 
140
  if __name__ == "__main__":