Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,9 @@ import random
|
|
3 |
import streamlit as st
|
4 |
from transformers import pipeline
|
5 |
|
|
|
|
|
|
|
6 |
def cleanup_output(text, prompt):
|
7 |
section = text[len(prompt):len(prompt) + 7]
|
8 |
print("Proposed Move: %s" % section)
|
|
|
3 |
import streamlit as st
|
4 |
from transformers import pipeline
|
5 |
|
6 |
+
generator2 = pipeline('text-generation', model='BlueSunflower/gpt2-medium-chess')
|
7 |
+
generator = pipeline('text-generation', model='gpt2')
|
8 |
+
|
9 |
def cleanup_output(text, prompt):
|
10 |
section = text[len(prompt):len(prompt) + 7]
|
11 |
print("Proposed Move: %s" % section)
|