Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from peft import PeftModel
|
|
10 |
|
11 |
# Constants
|
12 |
MAX_MAX_NEW_TOKENS = 2048
|
13 |
-
DEFAULT_MAX_NEW_TOKENS =
|
14 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
15 |
|
16 |
# Description and License Texts
|
@@ -53,7 +53,7 @@ class Story(Document):
|
|
53 |
|
54 |
# Utility function for prompts
|
55 |
def make_prompt(entry):
|
56 |
-
return f"### Human: {entry} ### Assistant:"
|
57 |
# f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS: {entry}"
|
58 |
|
59 |
# Gradio Function
|
|
|
10 |
|
11 |
# Constants
|
12 |
MAX_MAX_NEW_TOKENS = 2048
|
13 |
+
DEFAULT_MAX_NEW_TOKENS = 900
|
14 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
15 |
|
16 |
# Description and License Texts
|
|
|
53 |
|
54 |
# Utility function for prompts
|
55 |
def make_prompt(entry):
|
56 |
+
return f"### Human: Don't repeat the assesments, {entry} ### Assistant:"
|
57 |
# f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS: {entry}"
|
58 |
|
59 |
# Gradio Function
|