Duplicate from iricardoxd/bigscience-bloom-3b

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Gpt2
+emoji: 👀
+colorFrom: red
+colorTo: blue
+sdk: gradio
+sdk_version: 3.15.0
+app_file: app.py
+pinned: false
+license: cc-by-nc-2.0
+duplicated_from: iricardoxd/bigscience-bloom-3b
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,54 @@
+import torch
+from transformers import  AutoTokenizer, AutoModel, BloomTokenizerFast,BloomForCausalLM
+import gradio as gr
+modelo = 'bigscience/bloom-1b7' 
+tokenizer = AutoTokenizer.from_pretrained(modelo)
+model = BloomForCausalLM.from_pretrained(modelo)
+
+def generator(prompt,max_length, temp):
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
+    gen_tokens = model.generate(
+        input_ids,
+        do_sample=True,
+        temperature=temp,
+        max_length=max_length,
+    )
+    gen_text = tokenizer.batch_decode(gen_tokens)[0]
+    return gen_text
+
+
+
+
+
+
+def run(prompt, max_len, temp):
+    min_len = 1
+    output = generator(prompt,max_len, temp)
+    print(output)
+    return (output,"")
+
+if __name__ == "__main__":
+    demo = gr.Blocks()
+    with demo:
+        gr.Markdown(modelo)
+        with gr.Row():
+            with gr.Column():
+                text = gr.Textbox(
+                    label="Input",
+                    value=" ",  # should be set to " " when plugged into a real API
+                )
+                tokens = gr.Slider(1, 250, value=50, step=1, label="Tokens to generate")
+                temp = gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Temperature")
+
+                with gr.Row():
+                    submit = gr.Button("Submit")
+            with gr.Column():
+                text_error = gr.Markdown(label="Log information")
+                text_out = gr.Textbox(label="Output")
+        submit.click(
+            run,
+            inputs=[text, tokens, temp],
+            outputs=[text_out, text_error],
+        )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+torch
+transformers
+gradio