Update app.py

app.py

@@ -1,5 +1,4 @@
-#Choose the extractor. Both extractnet & dragnet have dependency conflicts with bertopic
-EXTRACTOR_NET = 'trafilatura'
+
 
 import numpy as np
 import onnxruntime
@@ -8,27 +7,23 @@ import gradio as gr
 import requests
 import json
 
-if(EXTRACTOR_NET == 'extractnet'):
-    try:
-        from extractnet import Extractor
-    except ImportError:
-        pass
-elif(EXTRACTOR_NET == 'dragnet'):
+
+try:
+    from extractnet import Extractor
+    EXTRACTOR_NET = 'extractnet'
+except ImportError:
     try:
         from dragnet import extract_content
+        EXTRACTOR_NET = 'dragnet'
     except ImportError:
-        pass
-elif(EXTRACTOR_NET == 'trafilatura'):
-    try:
-        import trafilatura
-    except ImportError:
-        pass
-else:
-    raise ImportError
+        try:
+            import trafilatura
+            EXTRACTOR_NET = 'trafilatura'
+        except ImportError:
+            raise ImportError
 
 print('[i] Using',EXTRACTOR_NET)
 
-from extractnet import Extractor
 import math
 from transformers import AutoTokenizer
 import spacy