upload all the files

app.py

@@ -1,19 +1,18 @@
 import os
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoProcessor
+from PIL import Image
+import torch
 
 def prompt_user_acceptance():
-    response = input("Do you accept the execution of remote code from the model repository? (y/N): ").strip().lower()
-    if response != 'y':
-        print("You must accept to continue.")
+    # Check environment variable
+    if os.getenv('TRUST_REMOTE_CODE') != 'true':
+        print("You must accept to continue. Set the environment variable TRUST_REMOTE_CODE=true.")
         exit(1)
 
 def main():
     prompt_user_acceptance()
 
-    import gradio as gr
-    from transformers import AutoModelForCausalLM, AutoProcessor
-    from PIL import Image
-    import torch
-
     # Load model and processor
     model = AutoModelForCausalLM.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", force_download=True)
     processor = AutoProcessor.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", force_download=True)