Spaces:
Build error
Build error
Update app_dialogue.py
Browse files- app_dialogue.py +20 -20
app_dialogue.py
CHANGED
@@ -554,7 +554,26 @@ def model_generation(
|
|
554 |
|
555 |
return generated_text
|
556 |
|
557 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
558 |
dope_callback = gr.CSVLogger()
|
559 |
dope_hf_callback = gr.HuggingFaceDatasetSaver(
|
560 |
hf_token=os.getenv("HF_AUTH_TOKEN"),
|
@@ -876,25 +895,6 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
|
|
876 |
print(f"chat_history (ELSE IF image is available) is -{chat_history}")
|
877 |
return "", None, chat_history
|
878 |
|
879 |
-
def process_example(message, image):
|
880 |
-
print("********* process_example **********")
|
881 |
-
clear_msg, image_value, chat = model_inference(
|
882 |
-
user_prompt_str=message,
|
883 |
-
chat_history=[],
|
884 |
-
image=image,
|
885 |
-
decoding_strategy="Greedy",
|
886 |
-
num_beams=None,
|
887 |
-
temperature=None,
|
888 |
-
no_repeat_ngram_size=None,
|
889 |
-
max_new_tokens=512,
|
890 |
-
min_length=16,
|
891 |
-
repetition_penalty=None,
|
892 |
-
length_penalty=None,
|
893 |
-
top_k=None,
|
894 |
-
top_p=0.95,
|
895 |
-
penalty_alpha=None,
|
896 |
-
)
|
897 |
-
return clear_msg, image_value, chat
|
898 |
|
899 |
textbox.submit(
|
900 |
fn=model_inference,
|
|
|
554 |
|
555 |
return generated_text
|
556 |
|
557 |
+
def process_example(message, image):
|
558 |
+
print("********* process_example **********")
|
559 |
+
clear_msg, image_value, chat = model_inference(
|
560 |
+
user_prompt_str=message,
|
561 |
+
chat_history=[],
|
562 |
+
image=image,
|
563 |
+
decoding_strategy="Greedy",
|
564 |
+
num_beams=None,
|
565 |
+
temperature=None,
|
566 |
+
no_repeat_ngram_size=None,
|
567 |
+
max_new_tokens=512,
|
568 |
+
min_length=16,
|
569 |
+
repetition_penalty=None,
|
570 |
+
length_penalty=None,
|
571 |
+
top_k=None,
|
572 |
+
top_p=0.95,
|
573 |
+
penalty_alpha=None,
|
574 |
+
)
|
575 |
+
return clear_msg, image_value, chat
|
576 |
+
|
577 |
dope_callback = gr.CSVLogger()
|
578 |
dope_hf_callback = gr.HuggingFaceDatasetSaver(
|
579 |
hf_token=os.getenv("HF_AUTH_TOKEN"),
|
|
|
895 |
print(f"chat_history (ELSE IF image is available) is -{chat_history}")
|
896 |
return "", None, chat_history
|
897 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
898 |
|
899 |
textbox.submit(
|
900 |
fn=model_inference,
|