NexusRaven2

Runtime error

Tonic commited on Jan 14, 2024

Commit

9c0da9e

verified ·

1 Parent(s): 0e5e7be

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gradio as gr
 from transformers import pipeline
 import os
@@ -17,6 +18,7 @@ raven_pipeline = pipeline(
     device_map="auto",
 )
 class DialogueToSpeechConverter:
     def __init__(self):
         self.raven_pipeline = raven_pipeline

+import spaces
 import gradio as gr
 from transformers import pipeline
 import os
     device_map="auto",
 )
+@spaces.GPU
 class DialogueToSpeechConverter:
     def __init__(self):
         self.raven_pipeline = raven_pipeline