Update app.py

app.py

@@ -30,8 +30,8 @@ def format_prompt(message, history):
 
 
 def chat_inf(system_prompt, prompt, history, temp, tokens, top_p, seed, client_choice):
-    #client = clients[int(client_choice) - 1]
-    client = clients[(client_choice)]
+    client = clients[int(client_choice) - 1]
+    
     if not history:
         history = []
         hist_len = 0
@@ -39,7 +39,7 @@ def chat_inf(system_prompt, prompt, history, temp, tokens, top_p, seed, client_c
         hist_len = len(history)
         print(hist_len)
 
-    sampler = keras_nlp.samplers.TopKSampler(k=5, seed=seed, top_p=top_p, temperature=temp)
+    sampler = keras_hub.samplers.TopKSampler(k=5, seed=seed, top_p=top_p, temperature=temp)
     client.compile(sampler=sampler)
     
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)