Spaces:
Build error
Build error
Zero GPU
Browse files
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
# -*- coding: utf-8 -*- Nour Eddine Zekaoui et al.
|
| 2 |
|
| 3 |
import torch
|
|
|
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
from peft import PeftModel
|
|
@@ -72,6 +73,7 @@ model = PeftModel.from_pretrained(
|
|
| 72 |
)
|
| 73 |
|
| 74 |
|
|
|
|
| 75 |
def generate(
|
| 76 |
instruction,
|
| 77 |
input=None,
|
|
|
|
| 1 |
# -*- coding: utf-8 -*- Nour Eddine Zekaoui et al.
|
| 2 |
|
| 3 |
import torch
|
| 4 |
+
import spaces
|
| 5 |
import gradio as gr
|
| 6 |
|
| 7 |
from peft import PeftModel
|
|
|
|
| 73 |
)
|
| 74 |
|
| 75 |
|
| 76 |
+
@spaces.GPU
|
| 77 |
def generate(
|
| 78 |
instruction,
|
| 79 |
input=None,
|