andreska commited on
Commit
3947df0
·
verified ·
1 Parent(s): 48097b8

Test switching to text-generation model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ file_path = "Adrega_P.I._User_Manual.docx"
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="Qwen/Qwen2.5-Coder-32B-Instruct")
 
18
 
19
  st.title("Adrega AI Help")
20
  #dataset = load_dataset("andreska/adregadocs", split="test")
 
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="Qwen/Qwen2.5-Coder-32B-Instruct")
18
+ pipe = pipeline("text2text-generation")
19
 
20
  st.title("Adrega AI Help")
21
  #dataset = load_dataset("andreska/adregadocs", split="test")