Spaces:

vitaliy-sharandin
/

WiseAI

Runtime error

vitaliy-sharandin commited on Nov 4, 2023

Commit

c06ec74

•

1 Parent(s): a1fb2a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-model = pipeline('text-generation', model='vitaliy-sharandin/wiseai')
 def generate_text(prompt):
     result = model(prompt, max_length=200)

 import gradio as gr
 from transformers import pipeline
+model = pipeline('text-generation', model='vitaliy-sharandin/wiseai', device=0)
 def generate_text(prompt):
     result = model(prompt, max_length=200)