subham1707 commited on
Commit
9bbe4b6
Β·
verified Β·
1 Parent(s): 4eb0f72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,10 +13,12 @@ import spaces
13
  @spaces.GPU
14
  def main():
15
  # Your existing Gradio app code here
 
16
  import gradio as gr
17
  from transformers import AutoTokenizer, AutoModelForCausalLM
18
  import torch
19
 
 
20
  model_id = "codellama/CodeLlama-7b-Instruct-hf"
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
  model = AutoModelForCausalLM.from_pretrained(
 
13
  @spaces.GPU
14
  def main():
15
  # Your existing Gradio app code here
16
+ import os
17
  import gradio as gr
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
  import torch
20
 
21
+ os.system("pip install gradio==3.50.2")
22
  model_id = "codellama/CodeLlama-7b-Instruct-hf"
23
  tokenizer = AutoTokenizer.from_pretrained(model_id)
24
  model = AutoModelForCausalLM.from_pretrained(