Update app.py

app.py

@@ -5,8 +5,6 @@ import json
 import random
 import os
 
-KEY = os.environ.get("KEY")
-
 base_endpoint = "https://api-inference.huggingface.co/models/"
 model_endpoint = f"{base_endpoint}HuggingFaceM4/idefics-80b-instruct"
 
@@ -31,12 +29,7 @@ default_pre_text = "(Describe the image) "
 def add_file(file):
     return file.name, gr.update(label='🖼️ Uploaded!')
     
-def predict(access_key, token, image, instructions, pre_input, input, seed = 42):  
-    
-    if (access_key != KEY):
-        print(">>> MODEL FAILED: Input: " + input + ", Attempted Key: " + access_key)
-        return ("[UNAUTHORIZED ACCESS]", input, []);
-
+def predict(token, image, instructions, pre_input, input, seed = 42):  
     if not instructions or len(instructions.strip()) <= 0:
         instructions = default_instructions
 
@@ -58,12 +51,7 @@ def predict(access_key, token, image, instructions, pre_input, input, seed = 42)
     )
     return response.strip()
 
-def predict_2(access_key, token, image, instructions, input, seed = 42):  
-    
-    if (access_key != KEY):
-        print(">>> MODEL FAILED: Input: " + input + ", Attempted Key: " + access_key)
-        return ("[UNAUTHORIZED ACCESS]", input, []);
-
+def predict_2(token, image, instructions, input, seed = 42):  
     formatted_input = instructions.replace("█", "<end_of_utterance>").replace("░", input).replace(symbol, f"![](https://statical-stc-itt.hf.space/file={image})")
     print(formatted_input)
     model = InferenceClient(model_endpoint, headers = { "Authorization": f"Bearer {token}" })
@@ -96,7 +84,6 @@ with gr.Blocks() as demo:
             text = gr.Textbox(label="Text Input", placeholder="Message...", lines=2)
             seed = gr.Slider( minimum = 0, maximum = 9007199254740991, value = 42, step = 1, interactive = True, label = "Seed" )
             token = gr.Textbox(label="Token", placeholder="Token...", lines=1)
-            access_key = gr.Textbox(label = "Access Key", lines = 1)
             
         with gr.Column():
             output = gr.Textbox(label = "Result", lines = 1)
@@ -104,8 +91,8 @@ with gr.Blocks() as demo:
             run2 = gr.Button("Generate Simple")
             maintain = gr.Button("☁️")
 
-    run.click(predict, inputs=[access_key, token, image, instructions, pre_text, text, seed], outputs=[output], queue = False)
-    run2.click(predict_2, inputs=[access_key, token, image, instructions, text, seed], outputs=[output], queue = False)
+    run.click(predict, inputs=[token, image, instructions, pre_text, text, seed], outputs=[output], queue = False)
+    run2.click(predict_2, inputs=[token, image, instructions, text, seed], outputs=[output], queue = False)
     maintain.click(cloud, inputs = [], outputs = [], queue = False)
 
 demo.launch()