amaltese commited on
Commit
ec1e648
·
verified ·
1 Parent(s): 46a9202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -10,8 +10,9 @@ wiki_wiki = wikipediaapi.Wikipedia(
10
  language='en'
11
  )
12
 
13
- # Load local AI model (Mistral-7B or Llama-2)
14
- chat_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 
15
 
16
  def get_wikipedia_summary(name):
17
  page = wiki_wiki.page(name)
 
10
  language='en'
11
  )
12
 
13
+ # Load local AI model (HuggingFaceH4/zephyr-7b-beta)
14
+ chat_model = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta")
15
+
16
 
17
  def get_wikipedia_summary(name):
18
  page = wiki_wiki.page(name)