ehristoforu commited on
Commit
e36667c
·
verified ·
1 Parent(s): b4b0ce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from threading import Thread
3
  from typing import Iterator
4
 
5
  import gradio as gr
6
- import spaces
7
  import torch
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
9
  from peft import AutoPeftModelForCausalLM
@@ -52,7 +52,7 @@ api.upload_folder(
52
  )
53
 
54
 
55
- @spaces.GPU()
56
  def generate(
57
  message: str,
58
  chat_history: list[dict],
 
3
  from typing import Iterator
4
 
5
  import gradio as gr
6
+ #import spaces
7
  import torch
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
9
  from peft import AutoPeftModelForCausalLM
 
52
  )
53
 
54
 
55
+ #@spaces.GPU()
56
  def generate(
57
  message: str,
58
  chat_history: list[dict],