Spaces:

UnionJack
/

CarlosElMejor

Sleeping

UnionJack commited on Oct 12, 2024

Commit

a937eb0

verified ·

1 Parent(s): 226f692

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,8 @@ pipe = pipeline(
     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
-    token = os.getenv('HF_TOKEN')
 )
 input = st.text_input("Ingrese el Promt")

     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
+    token = os.getenv('HF_TOKEN'),
+    max_lenght = 50
 )
 input = st.text_input("Ingrese el Promt")