Hasitha16 commited on
Commit
073de9b
·
verified ·
1 Parent(s): 571f6f2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +6 -1
model.py CHANGED
@@ -16,7 +16,12 @@ import re
16
 
17
  # === Pipelines ===
18
  summarizer = pipeline("summarization", model="google/pegasus-xsum")
19
- qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
 
 
 
 
20
  emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=1)
21
 
22
  # === Brief Summarization ===
 
16
 
17
  # === Pipelines ===
18
  summarizer = pipeline("summarization", model="google/pegasus-xsum")
19
+ qa_pipeline = pipeline(
20
+ "question-answering",
21
+ model="distilbert-base-cased-distilled-squad",
22
+ tokenizer="distilbert-base-cased-distilled-squad"
23
+ )
24
+
25
  emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=1)
26
 
27
  # === Brief Summarization ===