Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import transformers
|
|
|
|
| 3 |
|
| 4 |
title = """🙋🏻♂️Welcome to 🌟Tonic's 🤳🏻Phi-4 Demo"""
|
| 5 |
|
|
@@ -21,7 +22,7 @@ On 🌐Github: [Tonic-AI](https://github.com/tonic-ai) & contribute to🌟 [Dark
|
|
| 21 |
🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
|
| 22 |
"""
|
| 23 |
|
| 24 |
-
|
| 25 |
def generate_response(system_prompt, user_prompt, max_tokens, temperature):
|
| 26 |
pipeline = transformers.pipeline(
|
| 27 |
"text-generation",
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import transformers
|
| 3 |
+
import spaces
|
| 4 |
|
| 5 |
title = """🙋🏻♂️Welcome to 🌟Tonic's 🤳🏻Phi-4 Demo"""
|
| 6 |
|
|
|
|
| 22 |
🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
|
| 23 |
"""
|
| 24 |
|
| 25 |
+
@spaces.GPU
|
| 26 |
def generate_response(system_prompt, user_prompt, max_tokens, temperature):
|
| 27 |
pipeline = transformers.pipeline(
|
| 28 |
"text-generation",
|