Spaces:
Sleeping
Sleeping
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
2 |
+
import torch
|
3 |
+
import pandas as pd
|
4 |
+
import streamlit as st
|
5 |
+
from huggingface_hub import login
|
6 |
+
|
7 |
+
|
8 |
+
# Tu token secreto de Hugging Face
|
9 |
+
huggingface_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
|
10 |
+
|
11 |
+
|
12 |
+
model_name = "meta-llama/llama-3.1-8b-instruct" # Reemplaza con el modelo que quieras usar
|
13 |
+
model = AutoModelForCausalLM.from_pretrained(model_name)
|
14 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|