edusc182 commited on
Commit
da350e8
verified
1 Parent(s): 555c698

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoModel, AutoTokenizer
3
 
4
  # Carga tu modelo desde tu perfil de Hugging Face
5
- tokenizer = AutoTokenizer.from_pretrained("TuUsuario/TuModelo")
6
- model = AutoModel.from_pretrained("TuUsuario/TuModelo", use_safetensors=True)
7
 
8
  def predecir(texto):
9
  # Aqu铆 defines la l贸gica de inferencia de tu modelo
 
2
  from transformers import AutoModel, AutoTokenizer
3
 
4
  # Carga tu modelo desde tu perfil de Hugging Face
5
+ tokenizer = AutoTokenizer.from_pretrained("edusc182/Gemma_2B")
6
+ model = AutoModel.from_pretrained("edusc182/Gemma_2B", use_safetensors=True)
7
 
8
  def predecir(texto):
9
  # Aqu铆 defines la l贸gica de inferencia de tu modelo