Update app.py

app.py

@@ -1,7 +1,17 @@
 import gradio as gr
 from transformers import pipeline, AutoConfig
+import logging
 
-# Explicitly load the configuration with trust_remote_code
+# Temporary workaround: Extend the Logger class to include warning_once
+def warning_once(self, msg, *args, **kwargs):
+    if msg not in self._warned:
+        self.warning(msg, *args, **kwargs)
+        self._warned.add(msg)
+
+logging.Logger.warning_once = warning_once
+logging.Logger._warned = set()
+
+# Load the model configuration with trust_remote_code to execute local configuration
 config = AutoConfig.from_pretrained('tiiuae/falcon-40b-instruct', trust_remote_code=True)
 
 # Load the model once when the script starts using the loaded config