Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -553,13 +553,14 @@ def send_to_model_impl(prompt, model_selection, hf_model_choice, hf_custom_model
|
|
553 |
|
554 |
# Get the summary based on model selection
|
555 |
if model_selection == "HuggingFace Inference":
|
556 |
-
|
|
|
557 |
# Always try without API key first
|
558 |
summary = send_to_hf_inference(prompt, model_id)
|
559 |
if summary.startswith("Error: This model requires authentication") and hf_api_key:
|
560 |
# Only try with API key if the model specifically requires it
|
561 |
summary = send_to_hf_inference(prompt, model_id, hf_api_key, use_rate_limits)
|
562 |
-
|
563 |
elif model_selection == "Groq API":
|
564 |
if not groq_api_key:
|
565 |
return gr.HTML(""), "Error: Groq API key required", None
|
|
|
553 |
|
554 |
# Get the summary based on model selection
|
555 |
if model_selection == "HuggingFace Inference":
|
556 |
+
# Use the selected model ID directly
|
557 |
+
model_id = hf_custom_model if hf_model_choice == "Custom Model" else hf_model_choice
|
558 |
# Always try without API key first
|
559 |
summary = send_to_hf_inference(prompt, model_id)
|
560 |
if summary.startswith("Error: This model requires authentication") and hf_api_key:
|
561 |
# Only try with API key if the model specifically requires it
|
562 |
summary = send_to_hf_inference(prompt, model_id, hf_api_key, use_rate_limits)
|
563 |
+
|
564 |
elif model_selection == "Groq API":
|
565 |
if not groq_api_key:
|
566 |
return gr.HTML(""), "Error: Groq API key required", None
|