t

.gitattributes

@@ -1,27 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bin.* filter=lfs diff=lfs merge=lfs -text
-*.bz2 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
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack 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
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar 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
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zstandard filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,36 +1,36 @@
 
-# import gradio as gr
-# # from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
-# # prompt = "In a shocking finding, scientists discovered a herd of unicorns living in a remote, " \
-# #          "previously unexplored valley, in the Andes Mountains. Even more surprising to the " \
-# #          "researchers was the fact that the unicorns spoke perfect English."
+# prompt = "In a shocking finding, scientists discovered a herd of unicorns living in a remote, " \
+#          "previously unexplored valley, in the Andes Mountains. Even more surprising to the " \
+#          "researchers was the fact that the unicorns spoke perfect English."
 
 
-# def predict(text):
+def predict(text):
 
-#     tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
-#     model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
-#     input_ids = tokenizer(text, return_tensors="pt").input_ids
-#     gen_tokens = model.generate(input_ids, do_sample=True, temperature=0.9, max_length=100)
-#     gen_text = tokenizer.batch_decode(gen_tokens)[0]
-#     return gen_text
+    tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
+    model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
+    input_ids = tokenizer(text, return_tensors="pt").input_ids
+    gen_tokens = model.generate(input_ids, do_sample=True, temperature=0.9, max_length=100)
+    gen_text = tokenizer.batch_decode(gen_tokens)[0]
+    return gen_text
 
-# iface = gr.Interface(fn=predict, inputs="text", outputs="text")
-# iface.launch()
+iface = gr.Interface(fn=predict, inputs="text", outputs="text")
+iface.launch()
 
-# # all below works but testing
-# # import gradio as gr
+# all below works but testing
+# import gradio as gr
 
-# # title = "GPT-J-6B"
+# title = "GPT-J-6B"
 
-# # examples = [
-# #     ['The tower is 324 metres (1,063 ft) tall,'],
-# #     ["The Moon's orbit around Earth has"],
-# #     ["The smooth Borealis basin in the Northern Hemisphere covers 40%"]
-# # ]
+# examples = [
+#     ['The tower is 324 metres (1,063 ft) tall,'],
+#     ["The Moon's orbit around Earth has"],
+#     ["The smooth Borealis basin in the Northern Hemisphere covers 40%"]
+# ]
 
-# # gr.Interface.load("huggingface/EleutherAI/gpt-j-6B",
-# #     inputs=gr.inputs.Textbox(lines=10, label="Input Text"),
-# #     title=title, examples=examples).launch();
+# gr.Interface.load("huggingface/EleutherAI/gpt-j-6B",
+#     inputs=gr.inputs.Textbox(lines=10, label="Input Text"),
+#     title=title, examples=examples).launch();