File size: 530 Bytes
790ed47
 
49d62e1
dc398bf
 
790ed47
 
dc398bf
 
 
974dbb9
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["HF_TOKEN"]
login(huggingface_token)
#api_key = os.getenv("HF_TOKEN")
client = HuggingFaceHub(repo_id = 'mistralai/Mathstral-7B-v0.1') # 'mistralai/Mathstral-7B-v0.1')
prompt = "What is the best French cheese?"
response = client(prompt)
print(response)