Spaces:

Mr-Vicky-01
/

Webpage-Summarizer

Sleeping

Mr-Vicky-01 commited on May 10

Commit

1d3af5e

•

1 Parent(s): d998b3a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def get_answer(text):
     prompt = PromptTemplate(template=template, input_variables=["text"])
     llm_chain = LLMChain(prompt=prompt,
-                         llm=HuggingFaceHub(repo_id="Mr-Vicky-01/conversational_sumarization",
                                             model_kwargs={"max_length":100,
                                                           "max_new_tokens":100,
                                                           "do_sample": False}))

     prompt = PromptTemplate(template=template, input_variables=["text"])
     llm_chain = LLMChain(prompt=prompt,
+                         llm=HuggingFaceHub(repo_id="Mr-Vicky-01/Bart-Finetuned-conversational-summarization",
                                             model_kwargs={"max_length":100,
                                                           "max_new_tokens":100,
                                                           "do_sample": False}))