Tokymin commited on
Commit
2583ba2
1 Parent(s): 7f582f0

权限问题

Browse files
Files changed (2) hide show
  1. app.py +8 -0
  2. llama.js +9 -0
app.py CHANGED
@@ -1,10 +1,18 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
 
4
  # from huggingface_hub import login
5
  #
6
  # login()
7
  # 加载模型和tokenizer
 
 
 
 
 
 
 
8
  tokenizer = AutoTokenizer.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify_Model")
9
  model = AutoModelForSequenceClassification.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify_Model", num_labels=8)
10
  model.eval()
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
4
+ import os
5
  # from huggingface_hub import login
6
  #
7
  # login()
8
  # 加载模型和tokenizer
9
+ from transformers import AutoTokenizer, AutoModel
10
+
11
+ # Assuming you have set the HF_TOKEN environment variable with your Hugging Face token
12
+ huggingface_token = os.getenv('HF_TOKEN')
13
+
14
+ # Set up the token to use with the Hugging Face API
15
+ os.environ['HUGGINGFACE_CO_API_TOKEN'] = huggingface_token
16
  tokenizer = AutoTokenizer.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify_Model")
17
  model = AutoModelForSequenceClassification.from_pretrained("Tokymin/Mood_Anxiety_Disorder_Classify_Model", num_labels=8)
18
  model.eval()
llama.js ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import { AutoTokenizer } from '@xenova/transformers';
2
+
3
+ // Load tokenizer for a gated repository.
4
+ const tokenizer = await AutoTokenizer.from_pretrained('Tokymin/Mood_Anxiety_Disorder_Classify_Model');
5
+
6
+ // Encode text.
7
+ const text = 'Hello world!';
8
+ const encoded = tokenizer.encode(text);
9
+ console.log(encoded);