File size: 576 Bytes
790ed47
 
49d62e1
dc398bf
 
790ed47
 
dc398bf
977425f
dc398bf
154ac2a
0a421a5
4851603
 
 
f73e858
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import getpass
import os
from langchain import HuggingFaceHub
import streamlit as st
from huggingface_hub import login
#os.environ["MISTRAL_API_KEY"] = getpass.getpass()
#os.environ["HF_TOKEN"] = getpass.getpass()
# Token Secret of Hugging Face
huggingface_token = st.secrets["TF_TOKEN2"]
login(huggingface_token)
#api_key = os.getenv("HF_TOKEN2")
client = HuggingFaceHub(repo_id = 'mistralai/Mathstral-7B-v0.1', huggingfacehub_api_token=huggingface_token) # 'mistralai/Mathstral-7B-v0.1')
prompt = "What is the best French cheese?"
response = client(prompt)
print(response)