BramVanroy commited on
Commit
8f87350
·
verified ·
1 Parent(s): b4166e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from threading import Thread
5
  from typing import Iterator
6
 
7
  import gradio as gr
8
- import spaces
9
  import torch
10
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
11
 
@@ -19,7 +19,7 @@ if torch.cuda.is_available():
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
 
21
 
22
- @spaces.GPU
23
  def generate(
24
  message: str,
25
  chat_history: list[tuple[str, str]],
 
5
  from typing import Iterator
6
 
7
  import gradio as gr
8
+ # import spaces
9
  import torch
10
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
11
 
 
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
 
21
 
22
+ # @spaces.GPU
23
  def generate(
24
  message: str,
25
  chat_history: list[tuple[str, str]],