Merge pull request #1 from cdb24/milestone2

README.md

@@ -5,3 +5,12 @@ Tutorials Used:
 https://learn.microsoft.com/en-us/visualstudio/mac/docker-quickstart?view=vsmac-2022
 https://code.visualstudio.com/docs/containers/quickstart-python
 https://docs.docker.com/desktop/install/mac-install/
+
+title: Milestone 2
+emoji: 🐢
+colorFrom: green
+colorTo: purple
+sdk: streamlit
+sdk_version: 1.17.0
+app_file: app.py
+pinned: false

app.py

@@ -0,0 +1,13 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+
+model = AutoModelForCausalLM.from_pretrained("gpt2")
+
+from transformers import pipeline, set_seed
+
+generator = pipeline('text-generation', model='gpt2')
+
+set_seed(42)
+
+generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)

requirements.txt

@@ -1 +1,10 @@
 # To ensure app dependencies are ported from your virtual environment/host machine into your container, run 'pip freeze > requirements.txt' in the terminal to overwrite this file
+
+streamlit==0.84.1
+Pillow
+jax[cpu]
+flax
+transformers
+huggingface_hub
+googletrans==4.0.0-rc1
+protobuf==3.20