iamthewalrus67 commited on
Commit
130e53d
·
1 Parent(s): 76d5b12

Add installation step

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -2,6 +2,12 @@ import gradio as gr
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
 
 
 
 
 
 
5
  MODEL_ID = "le-llm/gemma-3-12b-it-reasoning"
6
 
7
  # Load model & tokenizer
@@ -56,4 +62,5 @@ chatbot = gr.ChatInterface(
56
  ],
57
  )
58
 
 
59
  chatbot.launch()
 
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
+ import os
6
+ import subprocess
7
+
8
+ def install_requirements():
9
+ subprocess.check_call([os.sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
10
+
11
  MODEL_ID = "le-llm/gemma-3-12b-it-reasoning"
12
 
13
  # Load model & tokenizer
 
62
  ],
63
  )
64
 
65
+ install_requirements()
66
  chatbot.launch()