Spaces:

huggingface
/

keras-chatbot-arena

Running on TPU v5e

martin-gorner HF staff commited on 29 days ago

Commit

b637f0b

•

1 Parent(s): 2ca0c5e

initial commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,8 +26,9 @@ for preset in model_presets:
     prompt, response = chat_state.send_message("Hello")
     print("model " + preset + "loaded and initialized.")
     print("The model responded: " + response)
-models = [load_model(preset) for preset in model_presets]
 # model = keras_hub.models.Llama3CausalLM.from_preset(
 #     "hf://meta-llama/Llama-3.2-1B-Instruct", dtype="bfloat16"
 # )

     prompt, response = chat_state.send_message("Hello")
     print("model " + preset + "loaded and initialized.")
     print("The model responded: " + response)
+    models.append(model)
+# For local debugging
 # model = keras_hub.models.Llama3CausalLM.from_preset(
 #     "hf://meta-llama/Llama-3.2-1B-Instruct", dtype="bfloat16"
 # )