DHEIVER commited on
Commit
1a92e57
·
verified ·
1 Parent(s): 9d4d900

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModel
2
+ import gradio as gr
3
+
4
+ # Carregar o modelo
5
+ model = AutoModel.from_pretrained("openbmb/MiniCPM-o-2_6", trust_remote_code=True)
6
+
7
+ # Função para gerar a resposta do modelo
8
+ def generate_response(input_text):
9
+ # Aqui você pode adicionar a lógica para processar o texto de entrada e gerar a resposta
10
+ # Por exemplo, se o modelo for um modelo de geração de texto, você pode usar model.generate()
11
+ # Como o modelo específico não foi detalhado, estou apenas retornando o texto de entrada como exemplo
12
+ response = f"Resposta do modelo para: {input_text}"
13
+ return response
14
+
15
+ # Criar a interface Gradio
16
+ interface = gr.Interface(
17
+ fn=generate_response, # Função que será chamada quando o usuário enviar o texto
18
+ inputs="text", # Tipo de entrada: texto
19
+ outputs="text", # Tipo de saída: texto
20
+ title="MiniCPM-o-2_6 Interface", # Título da interface
21
+ description="Digite um texto e veja a resposta do modelo MiniCPM-o-2_6." # Descrição
22
+ )
23
+
24
+ # Iniciar a interface
25
+ interface.launch()