aigorithm commited on
Commit
e418929
·
verified ·
1 Parent(s): a51eb3c

Upload 2 files

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,8 @@ import random
8
  import json
9
  from datetime import datetime
10
 
11
- chat_gen = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
 
12
 
13
  voice_options = {
14
  "Rachel": "EXAVITQu4vr4xnSDxMaL",
@@ -28,7 +29,7 @@ def generate_chat_bundle(prompt, character_inputs):
28
  f"Group chat between {', '.join(characters)}. Prompt: {prompt}. "
29
  "Use casual texting style: Name: message"
30
  )
31
- response = chat_gen(system_prompt, max_new_tokens=400)[0]["generated_text"]
32
  chat_lines = [(line.split(':', 1)[0].strip(), line.split(':', 1)[1].strip()) for line in response.splitlines() if ':' in line]
33
 
34
  html_chat = ""
@@ -108,5 +109,4 @@ with gr.Blocks() as app:
108
  generate_btn.click(fn=generate_chat_bundle, inputs=[prompt, characters], outputs=file_output)
109
  refresh_btn.click(fn=get_library, outputs=lib_html)
110
 
111
- # ✅ Required to launch on Hugging Face
112
  app.launch()
 
8
  import json
9
  from datetime import datetime
10
 
11
+ # Public-access model for Hugging Face
12
+ chat_gen = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
13
 
14
  voice_options = {
15
  "Rachel": "EXAVITQu4vr4xnSDxMaL",
 
29
  f"Group chat between {', '.join(characters)}. Prompt: {prompt}. "
30
  "Use casual texting style: Name: message"
31
  )
32
+ response = chat_gen(system_prompt, max_new_tokens=300)[0]["generated_text"]
33
  chat_lines = [(line.split(':', 1)[0].strip(), line.split(':', 1)[1].strip()) for line in response.splitlines() if ':' in line]
34
 
35
  html_chat = ""
 
109
  generate_btn.click(fn=generate_chat_bundle, inputs=[prompt, characters], outputs=file_output)
110
  refresh_btn.click(fn=get_library, outputs=lib_html)
111
 
 
112
  app.launch()