Inkcap commited on
Commit
5b0f2ea
1 Parent(s): 739be90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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)