Commit
•
bfb2b14
1
Parent(s):
dd3cf39
Add ZeroGPU to app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
# gr.load("models/kirankunapuli/Gemma-2B-Hinglish-LORA-v1.0").launch()
|
|
@@ -24,6 +25,7 @@ alpaca_prompt = """Below is an instruction that describes a task, paired with an
|
|
| 24 |
{}"""
|
| 25 |
|
| 26 |
|
|
|
|
| 27 |
def get_response(input_text: str) -> str:
|
| 28 |
inputs = tokenizer(
|
| 29 |
[
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# gr.load("models/kirankunapuli/Gemma-2B-Hinglish-LORA-v1.0").launch()
|
|
|
|
| 25 |
{}"""
|
| 26 |
|
| 27 |
|
| 28 |
+
@spaces.GPU
|
| 29 |
def get_response(input_text: str) -> str:
|
| 30 |
inputs = tokenizer(
|
| 31 |
[
|