Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from gradio_client import Client
-from huggingface_hub import InferenceClient, HFRepository
+from huggingface_hub import InferenceClient
 
 import random
 
@@ -21,23 +21,6 @@ for model in models:
     except Exception as e:
         print(f"Error loading model {model}: {str(e)}")
 
-# Cargar conjuntos de datos desde un archivo de texto
-def load_datasets_from_file(filename):
-    with open(filename, 'r') as file:
-        datasets = [line.strip() for line in file.readlines() if line.strip()]
-    return datasets
-
-# Cargar conjuntos de datos desde un archivo de texto (datasets.txt)
-datasets = load_datasets_from_file("datasets.txt")
-
-# Crear clientes de repositorio para cada conjunto de datos
-repository_clients = []
-for dataset in datasets:
-    try:
-        repository_clients.append(HFRepository(dataset))
-    except Exception as e:
-        print(f"Error loading dataset {dataset}: {str(e)}")
-
 VERBOSE = False
 
 def format_prompt(message, history, cust_p):
@@ -52,8 +35,8 @@ def format_prompt(message, history, cust_p):
     return prompt
 
 def chat_inf(system_prompt, prompt, history, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, cust_p):
-    if not clients or not repository_clients:
-        yield [("Error", "No models or datasets available")], memory
+    if not clients:
+        yield [("Error", "No models available")], memory
     else:
         try:
             client = clients[int(client_choice) - 1]