amaltese commited on
Commit
db135a5
·
verified ·
1 Parent(s): 46b622e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  import wikipediaapi
6
 
7
  # Custom User-Agent to avoid API blocking
8
- USER_AGENT = "HistoricalChatbot/1.0 (Contact: your-email@example.com)"
9
 
10
  # Load Wikipedia API with custom headers
11
  wiki_wiki = wikipediaapi.Wikipedia(
@@ -14,7 +14,7 @@ wiki_wiki = wikipediaapi.Wikipedia(
14
  )
15
 
16
  # Load local AI model (Mistral-7B or Llama-2)
17
- chat_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
18
 
19
  def get_wikipedia_summary(name):
20
  page = wiki_wiki.page(name)
 
5
  import wikipediaapi
6
 
7
  # Custom User-Agent to avoid API blocking
8
+ USER_AGENT = "HistoricalChatbot/1.0 (Contact: amaltese@indiana.edu)"
9
 
10
  # Load Wikipedia API with custom headers
11
  wiki_wiki = wikipediaapi.Wikipedia(
 
14
  )
15
 
16
  # Load local AI model (Mistral-7B or Llama-2)
17
+ chat_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
18
 
19
  def get_wikipedia_summary(name):
20
  page = wiki_wiki.page(name)