Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import os
|
|
9 |
from spaces import GPU
|
10 |
from dotenv import load_dotenv
|
11 |
|
12 |
-
load_dotenv()
|
13 |
|
14 |
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
|
15 |
|
@@ -88,7 +88,7 @@ def remove_duplicates(text):
|
|
88 |
seen_lines.add(line)
|
89 |
return '\n'.join(unique_lines)
|
90 |
|
91 |
-
@GPU(duration=1)
|
92 |
def generate_model_response(model, inputs):
|
93 |
try:
|
94 |
response = model(inputs)
|
|
|
9 |
from spaces import GPU
|
10 |
from dotenv import load_dotenv
|
11 |
|
12 |
+
load_dotenv()
|
13 |
|
14 |
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
|
15 |
|
|
|
88 |
seen_lines.add(line)
|
89 |
return '\n'.join(unique_lines)
|
90 |
|
91 |
+
@GPU(duration=1)
|
92 |
def generate_model_response(model, inputs):
|
93 |
try:
|
94 |
response = model(inputs)
|