Update app_dialogue.py

app_dialogue.py

@@ -12,7 +12,9 @@ import gradio as gr
 import PIL
 from gradio import processing_utils
 from gradio_client.client import DEFAULT_TEMP_DIR
-from text_generation import Client
+from huggingface_hub import InferenceClient
+
+#from text_generation import Client
 from transformers import AutoProcessor
 
 
@@ -505,12 +507,14 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
             current_image=image,
             history=chat_history,
         )
+        client = InferenceClient("HuggingFaceM4/idefics-80b-instruct")
 
+        '''
         client_endpoint = API_PATHS[model_selector]
         client = Client(
             base_url=client_endpoint,
             headers={"x-use-cache": "0", "Authorization": f"Bearer {API_TOKEN}"},
-        )
+        )'''
 
         # Common parameters to all decoding strategies
         # This documentation is useful to read: https://huggingface.co/docs/transformers/main/en/generation_strategies