andreska commited on
Commit
d9f5ada
·
verified ·
1 Parent(s): 2df308e

Updated model to llama

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ file_path = "Adrega_P.I._User_Manual.docx"
13
 
14
  #pipe = pipeline("question-answering")
15
  #pipe = pipeline("text-generation")
16
- pipe = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
17
 
18
  st.title("Adrega AI Help")
19
  #dataset = load_dataset("andreska/adregadocs", split="test")
 
13
 
14
  #pipe = pipeline("question-answering")
15
  #pipe = pipeline("text-generation")
16
+ #pipe = pipeline("question-answering", model="deepset/roberta-base-squad2")
17
+ pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
18
 
19
  st.title("Adrega AI Help")
20
  #dataset = load_dataset("andreska/adregadocs", split="test")