Merge branch 'main' of https://huggingface.co/spaces/Silas-Blanchard/Chess_LLM_Space

app.py

@@ -1,11 +1,8 @@
 import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
-
 from transformers import pipeline
-generator2 = pipeline('text-generation', model='meta-llama/Llama-2-7b-chat-hf')
+generator2 = pipeline('text-generation', model='gpt2')
 
 prompt = "Return an opening move in chess."
 generated_text = generator2(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
-print(generated_text)
+st.write(generated_text)

requirements.txt

@@ -3,4 +3,5 @@ python-chess==1.1.0
 matplotlib==3.8.2
 requests==2.28.2
 pandas==1.5.3
-transformers==4.36.2
+transformers==4.36.2
+torch==2.1.2