zerogpu

app.py

@@ -3,7 +3,7 @@
 import os
 from threading import Thread
 from typing import Iterator
-
+import spaces
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer