Spaces:
Running
Running
feat: type error
Browse files
app.py
CHANGED
|
@@ -3,7 +3,6 @@ import gradio as gr
|
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
pipe = pipeline("feature-extraction", model="jinaai/jina-embeddings-v2-base-code", trust_remote_code=True)
|
| 6 |
-
pipe.to('cuda')
|
| 7 |
|
| 8 |
@spaces.GPU
|
| 9 |
def generate(prompt):
|
|
|
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
pipe = pipeline("feature-extraction", model="jinaai/jina-embeddings-v2-base-code", trust_remote_code=True)
|
|
|
|
| 6 |
|
| 7 |
@spaces.GPU
|
| 8 |
def generate(prompt):
|