data-ject commited on
Commit
1db7c5c
·
verified ·
1 Parent(s): cc3bc69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -15,7 +15,7 @@ if not api_token:
15
  raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
16
 
17
  # Load model and tokenizer from Hugging Face Model Hub with authentication
18
- model_name = "meta-llama/Meta-Llama-3.1-70B-Instruct"
19
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
20
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
21
 
@@ -70,10 +70,9 @@ iface = gr.Interface(
70
  inputs="text",
71
  outputs="text",
72
  title="Meta-Llama Chatbot",
73
- description="A chatbot powered by the Meta-Llama-3.1-70B-Instruct model."
74
  )
75
 
76
  # Launch the interface
77
  if __name__ == "__main__":
78
  iface.launch()
79
-
 
15
  raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
16
 
17
  # Load model and tokenizer from Hugging Face Model Hub with authentication
18
+ model_name = "meta-llama/Meta-Llama-3-70B-Instruct"
19
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
20
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
21
 
 
70
  inputs="text",
71
  outputs="text",
72
  title="Meta-Llama Chatbot",
73
+ description="A chatbot powered by the Meta-Llama-3-70B-Instruct model."
74
  )
75
 
76
  # Launch the interface
77
  if __name__ == "__main__":
78
  iface.launch()