smhavens commited on
Commit
5d26793
·
1 Parent(s): 3854d90

Bug fix again, fixed prompt and made answer global

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,7 +13,7 @@ import evaluate
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
15
 
16
- answer = "Pizza"
17
 
18
 
19
  #Mean Pooling - Take attention mask into account for correct averaging
@@ -114,10 +114,11 @@ def main():
114
  word1 = "Black"
115
  word2 = "White"
116
  word3 = "Sun"
 
117
  answer = "Moon"
118
  guesses = []
119
 
120
- prompt = "{word1} is to {word2} as {word3} is to ____"
121
  with gr.Blocks() as iface:
122
  gr.Markdown(prompt)
123
  with gr.Tab("Guess"):
 
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
15
 
16
+ # answer = "Pizza"
17
 
18
 
19
  #Mean Pooling - Take attention mask into account for correct averaging
 
114
  word1 = "Black"
115
  word2 = "White"
116
  word3 = "Sun"
117
+ global answer
118
  answer = "Moon"
119
  guesses = []
120
 
121
+ prompt = f"{word1} is to {word2} as {word3} is to ____"
122
  with gr.Blocks() as iface:
123
  gr.Markdown(prompt)
124
  with gr.Tab("Guess"):