Hugging Face's logo

Spaces:
enotkrutoy
/
WhiteRabbitNeo-Llama-3-WhiteRabbitNeo-8B-v2.0
Runtime error

App Files Community
enotkrutoy commited on
Commit
de7ac8a
•
1 Parent(s): ce1c1cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
- # Идентификатор модели
5
- model_id = "WhiteRabbitNeo/WhiteRabbitNeo-13B-v1"
6
 
 
 
7
 
8
 
9
  # Создание клиента для доступа к модели
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
+ # Load model directly
5
+ from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("WhiteRabbitNeo/Trinity-33B-v1.0")
8
+ model = AutoModelForCausalLM.from_pretrained("WhiteRabbitNeo/Trinity-33B-v1.0")
9
 
10
 
11
  # Создание клиента для доступа к модели