yasserrmd commited on
Commit
bb262ae
1 Parent(s): 226b6dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
- import spaces
5
 
6
  model_name = "MBZUAI-Paris/Atlas-Chat-9B"
7
  dtype = torch.bfloat16
@@ -13,7 +13,7 @@ model = AutoModelForCausalLM.from_pretrained(
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
 
15
 
16
- @spaces.GPU
17
  def chat(input_text, history=[]):
18
  # Tokenize the input and generate response
19
  inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
+ #import spaces
5
 
6
  model_name = "MBZUAI-Paris/Atlas-Chat-9B"
7
  dtype = torch.bfloat16
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
 
15
 
16
+ #@spaces.GPU
17
  def chat(input_text, history=[]):
18
  # Tokenize the input and generate response
19
  inputs = tokenizer(input_text, return_tensors="pt").to(model.device)