Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 from threading import Thread
 from typing import Generator
 from liger_kernel.transformers import apply_liger_kernel_to_llama
+from gradio_client import Client
 
 
 
@@ -374,6 +375,22 @@ with gr.Blocks() as demo:
 
 	gr.Markdown(DESCRIPTION)
 
+def extract_x_ip_token(request):
+    # Pobierz token z nagłówków żądania
+    x_ip_token = request.headers.get('x-ip-token')
+    if x_ip_token:
+        # Inicjalizuj klienta Gradio z tokenem
+        client = Client("Twoja/Przestrzen", headers={"X-IP-Token": x_ip_token})
+        # Wykonaj zapytanie do modelu
+        result = client.predict(input_data="Twoje dane wejściowe")
+        return result
+    else:
+        return "Token X-IP-Token nie został znaleziony."
+
+with gr.Blocks() as demo:
+    input_data = gr.Textbox(label="Wprowadź dane")
+    output_data = gr.Textbox(label="Wynik")
+    input_data.submit(extract_x_ip_token, inputs=[input_data], outputs=[output_data])
 
 if __name__ == "__main__":
     demo.launch(show_error=True)