Update app.py
Browse files
app.py
CHANGED
@@ -12,12 +12,11 @@ llm = Llama(
|
|
12 |
|
13 |
def chat(message, history):
|
14 |
system_prompt = """Ti si YugoGPT, AI asistent koji prvenstveno komunicira na srpskom jeziku.
|
15 |
-
Tvoj zadatak je da daješ detaljne
|
16 |
|
17 |
full_prompt = f"""SYSTEM: {system_prompt}
|
18 |
USER: {message}
|
19 |
-
ASSISTANT:
|
20 |
-
"""
|
21 |
|
22 |
response = llm(
|
23 |
full_prompt,
|
@@ -48,9 +47,10 @@ demo = gr.ChatInterface(
|
|
48 |
)
|
49 |
|
50 |
if __name__ == "__main__":
|
51 |
-
demo.launch(
|
52 |
server_name="0.0.0.0",
|
53 |
server_port=7860,
|
54 |
share=False
|
55 |
)
|
56 |
|
|
|
|
12 |
|
13 |
def chat(message, history):
|
14 |
system_prompt = """Ti si YugoGPT, AI asistent koji prvenstveno komunicira na srpskom jeziku.
|
15 |
+
Tvoj zadatak je da daješ detaljne i korisne odgovore na srpskom jeziku."""
|
16 |
|
17 |
full_prompt = f"""SYSTEM: {system_prompt}
|
18 |
USER: {message}
|
19 |
+
ASSISTANT: """
|
|
|
20 |
|
21 |
response = llm(
|
22 |
full_prompt,
|
|
|
47 |
)
|
48 |
|
49 |
if __name__ == "__main__":
|
50 |
+
demo.queue().launch(
|
51 |
server_name="0.0.0.0",
|
52 |
server_port=7860,
|
53 |
share=False
|
54 |
)
|
55 |
|
56 |
+
|