Spaces:
Running
Running
update code
Browse files
app.py
CHANGED
|
@@ -23,8 +23,8 @@ model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
|
|
| 23 |
|
| 24 |
|
| 25 |
# to summarize
|
| 26 |
-
|
| 27 |
-
summarizer = pipeline('text2text-generation', model='describeai/gemini')
|
| 28 |
#nlpaueb/legal-bert-base-uncased
|
| 29 |
|
| 30 |
|
|
@@ -65,17 +65,17 @@ async def ppt_content(data):
|
|
| 65 |
|
| 66 |
|
| 67 |
|
| 68 |
-
result_1 = agent.run_sync(user_prompt =
|
| 69 |
|
| 70 |
print(result_1.data)
|
| 71 |
|
| 72 |
|
| 73 |
def ai_ppt(data):
|
| 74 |
#call summerizer to summerize pdf
|
| 75 |
-
summary = summarizer("
|
| 76 |
|
| 77 |
-
|
| 78 |
-
summary_texts = [item['generated_text'] for item in summary]
|
| 79 |
asyncio.run(ppt_content(data=summary_texts))
|
| 80 |
|
| 81 |
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
# to summarize
|
| 26 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
| 27 |
+
#summarizer = pipeline('text2text-generation', model='describeai/gemini')
|
| 28 |
#nlpaueb/legal-bert-base-uncased
|
| 29 |
|
| 30 |
|
|
|
|
| 65 |
|
| 66 |
|
| 67 |
|
| 68 |
+
result_1 = agent.run_sync(user_prompt = data)
|
| 69 |
|
| 70 |
print(result_1.data)
|
| 71 |
|
| 72 |
|
| 73 |
def ai_ppt(data):
|
| 74 |
#call summerizer to summerize pdf
|
| 75 |
+
summary = summarizer("".join(data), max_length=400, min_length=100, truncation=True,do_sample=False)
|
| 76 |
|
| 77 |
+
summary_texts = [item['summary_text'] for item in summary]
|
| 78 |
+
#summary_texts = [item['generated_text'] for item in summary]
|
| 79 |
asyncio.run(ppt_content(data=summary_texts))
|
| 80 |
|
| 81 |
|