kwabs22 commited on
Commit
69fb1b1
·
1 Parent(s): 0ebff8f

prompt edits

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,7 +46,7 @@ def generate_response(user_message): #generate_response_token_by_token
46
  if token_buffer:
47
  elapsed_time = time.time() - start_time # Calculate elapsed time
48
  alltokens += token_buffer
49
- yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { tokencount / elapsed_time}]"
50
 
51
  finally:
52
  try:
@@ -67,15 +67,15 @@ def generate_response(user_message): #generate_response_token_by_token
67
 
68
 
69
  def custom_generate_response(cust_user_message):
70
- cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message + '\n\n'
71
  yield from generate_response(cust_user_message)
72
 
73
  def custom_generate_response1(cust_user_message):
74
- cust_user_message = CustomPrompts[1] + '\n\n' + cust_user_message + '\n\n'
75
  yield from generate_response(cust_user_message)
76
 
77
  def custom_generate_response2(cust_user_message):
78
- cust_user_message = CustomPrompts[2] + '\n' + cust_user_message + '\n\n'
79
  yield from generate_response(cust_user_message)
80
 
81
  CustomPrompts = [
 
46
  if token_buffer:
47
  elapsed_time = time.time() - start_time # Calculate elapsed time
48
  alltokens += token_buffer
49
+ yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { tokencount / elapsed_time }]"
50
 
51
  finally:
52
  try:
 
67
 
68
 
69
  def custom_generate_response(cust_user_message):
70
+ cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message + '\n\nClass Diagram:'
71
  yield from generate_response(cust_user_message)
72
 
73
  def custom_generate_response1(cust_user_message):
74
+ cust_user_message = CustomPrompts[1] + '\n\n' + cust_user_message + '\n\nPydot Code:'
75
  yield from generate_response(cust_user_message)
76
 
77
  def custom_generate_response2(cust_user_message):
78
+ cust_user_message = CustomPrompts[2] + '\n' + cust_user_message + '\n\nScene Details'
79
  yield from generate_response(cust_user_message)
80
 
81
  CustomPrompts = [