Update main.py

main.py

@@ -5,7 +5,7 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import re
 import replicate
-import OSError
+import os
 from supabase import create_client, Client
 
 url: str = os.environ.get("DB_URL")
@@ -118,7 +118,7 @@ async def root(site: str = 'abcnews.go'):
     try:
 
         response = supabase.table('news').select("*")
-        .eq('source', 'www.'+ site + '.com').is_('image_url', 'null')
+        .eq('source', f'www.{site}.com').is_('image_url', 'null')
         .order('published_date', desc=True).limit(15).execute()