1 files changed, 6 insertions, 1 deletions
diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py
index cd9a971c..cbbbd75e 100644
--- a/g4f/Provider/helper.py
+++ b/g4f/Provider/helper.py
@@ -17,12 +17,17 @@ from browser_cookie3 import (
     firefox,
     BrowserCookieError
 )
+try: 
+     from selenium.webdriver.remote.webdriver import WebDriver 
+ except ImportError: 
+     class WebDriver(): 
+         pass
 try:
     from undetected_chromedriver import Chrome, ChromeOptions
 except ImportError:
     class Chrome():
         def __init__():
-            raise RuntimeError('Please install "undetected_chromedriver" and "pyvirtualdisplay" package')
+            raise RuntimeError('Please install the "undetected_chromedriver" package')
     class ChromeOptions():
         def add_argument():
             pass