remove llama

app.py

@@ -4,8 +4,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 vicuna_model = AutoModelForCausalLM.from_pretrained("lmsys/vicuna-7b-v1.3")
 vicuna_tokenizer = AutoTokenizer.from_pretrained("lmsys/vicuna-7b-v1.3")
 
-llama_model = AutoModelForCausalLM.from_pretrained("luodian/llama-7b-hf")
-llama_tokenizer = AutoTokenizer.from_pretrained("luodian/llama-7b-hf")
+# llama_model = AutoModelForCausalLM.from_pretrained("luodian/llama-7b-hf")
+# llama_tokenizer = AutoTokenizer.from_pretrained("luodian/llama-7b-hf")
 
 # Define the function for generating responses
 def generate_response(model, tokenizer, prompt):
@@ -17,9 +17,9 @@ def generate_response(model, tokenizer, prompt):
 # Define the Gradio interface
 def chatbot_interface(prompt):
     vicuna_response = generate_response(vicuna_model, vicuna_tokenizer, prompt)
-    llama_response = generate_response(llama_model, llama_tokenizer, prompt)
+    # llama_response = generate_response(llama_model, llama_tokenizer, prompt)
 
-    return {"Vicuna-7B": vicuna_response, "Llama-7B": llama_response}
+    return {"Vicuna-7B": vicuna_response}
 
 iface = gr.Interface(fn=chatbot_interface, 
                      inputs="text",