Spaces:

Walid-Ahmed
/

Text_generation

Running on Zero

Walid-Ahmed commited on 29 days ago

Commit

08ef01f

verified ·

1 Parent(s): 2380ea8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
-import space
 def load_model(model_name):
     device = "cuda" if torch.cuda.is_available() else "cpu"

 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
+import spaces
 def load_model(model_name):
     device = "cuda" if torch.cuda.is_available() else "cpu"