Upload app.py

app.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 from duckduckgo_search import DDGS
 from newspaper import Article
 import scipy
+import random
 from transformers import (
     MT5Tokenizer,
     AdamW,
@@ -22,7 +23,7 @@ class Webapp:
     self.DEVICE = 0 if torch.cuda.is_available() else "cpu"
     self.REF_MODEL = 'google/mt5-small'
     self.MODEL_NAME = 'Ahmedasd/arabic-summarization-hhh-100-batches'
-    self.model_id = "openai/whisper-base"
+    self.model_id = "openai/whisper-tiny"
     self.tts_model_id = "SeyedAli/Arabic-Speech-synthesis"
     self.tts_model = VitsModel.from_pretrained(self.tts_model_id).to(self.DEVICE)
     self.tts_tokenizer = AutoTokenizer.from_pretrained(self.tts_model_id)
@@ -57,7 +58,14 @@ class Webapp:
         urls = [r['url'] for r in results]
         print('successful connection!')
       except Exception as error:
-        urls = ['https://www.bbc.com/arabic/media-65576589']
+        examples = ['https://www.bbc.com/arabic/media-65576589',
+                    'https://www.bbc.com/arabic/articles/czr8dk93231o',
+                    'https://www.bbc.com/arabic/articles/c0jyd2yweplo',
+                    'https://www.bbc.com/arabic/articles/cnd8wwdyyzko',
+                    'https://www.bbc.com/arabic/articles/c3gyxymp0z1o',
+                    'https://www.bbc.com/arabic/articles/c3g28kl8zj4o'
+                    ]
+        urls = [random.choice(examples)]
       
     articles = []
     for url in urls: