Nolan Higgins
commited on
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from langchain_huggingface import HuggingFaceEndpoint
|
| 2 |
+
import streamlit as st
|
| 3 |
+
model_id="mistralai/Mistral-7B-Instruct-v0.3"
|
| 4 |
+
def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.1):
|
| 5 |
+
llm = HuggingFaceEndpoint(
|
| 6 |
+
repo_id=model_id,
|
| 7 |
+
max_new_tokens=max_new_tokens,
|
| 8 |
+
temperature=temperature,
|
| 9 |
+
token = os.getenv("HF_TOKEN")
|
| 10 |
+
)
|
| 11 |
+
return llm
|