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)
|