pablot21 commited on
Commit
6a3db50
1 Parent(s): 5267f07

Agrego application file

Browse files
Files changed (1) hide show
  1. app.py +56 -0
app.py ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import requests
3
+ from huggingface.hub import load_model
4
+
5
+ # Define la funci贸n para procesar el documento con el prompt
6
+ def process_document(document, prompt, model_name):
7
+ # Carga el modelo
8
+ model = load_model(model_name)
9
+
10
+ # Prepara el texto de entrada
11
+ input_text = document + prompt
12
+
13
+ # Genera la respuesta del modelo
14
+ response = model(input_text)
15
+
16
+ # Extrae la respuesta
17
+ model_response = response["generated_text"][0]
18
+
19
+ # Devuelve la respuesta del modelo
20
+ return model_response
21
+
22
+ # Crea la interfaz de usuario de Streamlit
23
+ st.title("Aplicaci贸n de Procesamiento de Documentos con Prompt")
24
+
25
+ # Campo de texto para el documento
26
+ document_input = st.text_area("Ingrese el documento:")
27
+
28
+ # Campo de texto para el prompt
29
+ prompt_input = st.text_area("Ingrese el prompt:")
30
+
31
+ # Selector de modelo
32
+ model_options = ["nombre_modelo_1", "nombre_modelo_2", ...]
33
+ selected_model = st.selectbox("Seleccione un modelo:", model_options)
34
+
35
+ # Bot贸n para procesar el documento
36
+ if st.button("Procesar"):
37
+ # Procesa el documento con el prompt y el modelo seleccionado
38
+ model_response = process_document(document_input, prompt_input, selected_model)
39
+
40
+ # Muestra la respuesta del modelo
41
+ st.write("**Respuesta del modelo:**")
42
+ st.write(model_response)
43
+
44
+ # Importa las librer铆as necesarias para el despliegue en Hugging Face Spaces
45
+ from huggingface_hub import push_to_hub
46
+
47
+ # Define la configuraci贸n de la aplicaci贸n
48
+ app_config = {
49
+ "title": "Aplicaci贸n de Procesamiento de Documentos con Prompt",
50
+ "description": "Esta aplicaci贸n permite ingresar un documento, un prompt y enviarlos a un modelo para obtener una respuesta.",
51
+ "keywords": ["procesamiento de lenguaje natural", "prompt", "modelo"],
52
+ "category": "apps",
53
+ }
54
+
55
+ # Despliega la aplicaci贸n en Hugging Face Spaces (descomentar para desplegar)
56
+ push_to_hub(app=app, config=app_config)