nisten commited on
Commit
b3ca2da
1 Parent(s): e659cfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import spaces
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
  import subprocess
 
6
 
7
  # Force install the specific transformers version from the GitHub PR
8
  subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "--no-deps", "git+https://github.com/Muennighoff/transformers.git@olmoe"])
@@ -39,7 +40,7 @@ def generate_response(message, history):
39
  return response.strip()
40
 
41
  with gr.Blocks() as demo:
42
- gr.Markdown("# Pissed Off Karpathy Chatbot")
43
  chatbot = gr.Chatbot()
44
  msg = gr.Textbox()
45
  clear = gr.Button("Clear")
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
  import subprocess
6
+ import sys
7
 
8
  # Force install the specific transformers version from the GitHub PR
9
  subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "--no-deps", "git+https://github.com/Muennighoff/transformers.git@olmoe"])
 
40
  return response.strip()
41
 
42
  with gr.Blocks() as demo:
43
+ gr.Markdown("#Karpathy Chatbot")
44
  chatbot = gr.Chatbot()
45
  msg = gr.Textbox()
46
  clear = gr.Button("Clear")