Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,15 +19,16 @@ model = model.to("cuda").eval()
|
|
| 19 |
|
| 20 |
|
| 21 |
@spaces.GPU(duration=10)
|
| 22 |
-
def greet(
|
|
|
|
| 23 |
lm = Transformers(model=model, tokenizer=tok)
|
| 24 |
-
lm +=
|
| 25 |
lm += gen(
|
| 26 |
"response",
|
| 27 |
n=1,
|
| 28 |
temperature=0,
|
| 29 |
max_tokens=1024,
|
| 30 |
-
regex=
|
| 31 |
)
|
| 32 |
return lm["response"]
|
| 33 |
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
@spaces.GPU(duration=10)
|
| 22 |
+
def greet(prompt):
|
| 23 |
+
temp = f"{prompt} Only use results from the genomic data commons in your response and provide frequencies as a percentage. Only report the final response."
|
| 24 |
lm = Transformers(model=model, tokenizer=tok)
|
| 25 |
+
lm += temp
|
| 26 |
lm += gen(
|
| 27 |
"response",
|
| 28 |
n=1,
|
| 29 |
temperature=0,
|
| 30 |
max_tokens=1024,
|
| 31 |
+
regex="The final answer is: \d*\.\d*%",
|
| 32 |
)
|
| 33 |
return lm["response"]
|
| 34 |
|