skavya commited on
Commit
b5d9e89
1 Parent(s): 5cb0d9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,8 +3,11 @@ import transformers
3
  import youtube_transcript_api
4
  from transformers import pipeline
5
  from youtube_transcript_api import YouTubeTranscriptApi
 
6
 
7
  summarizer = pipeline("summarization")
 
 
8
 
9
  def greet(link):
10
  try:
 
3
  import youtube_transcript_api
4
  from transformers import pipeline
5
  from youtube_transcript_api import YouTubeTranscriptApi
6
+ from datasets import Dataset
7
 
8
  summarizer = pipeline("summarization")
9
+ summarizer=Dataset.from_dict({"data": summarizer})
10
+ summarizer=summarizer.with_format("torch")
11
 
12
  def greet(link):
13
  try: