Spaces:
Runtime error
Runtime error
added integration for the hallucination model
Browse files
app.py
CHANGED
@@ -15,7 +15,18 @@ import time
|
|
15 |
dotenv.load_dotenv()
|
16 |
|
17 |
seamless_client = Client("facebook/seamless_m4t")
|
|
|
18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
|
20 |
|
21 |
|
@@ -425,9 +436,16 @@ with gr.Blocks(theme='ParityError/Anime') as iface :
|
|
425 |
audio_input = gr.Audio(label="speak",type="filepath",sources="microphone")
|
426 |
audio_output = gr.Markdown(label="output text")
|
427 |
audio_button = gr.Button("process audio")
|
|
|
|
|
|
|
|
|
|
|
428 |
text_button.click(process_and_query, inputs=text_input, outputs=text_output)
|
429 |
image_button.click(process_image, inputs=image_input, outputs=image_output)
|
430 |
audio_button.click(process_speech, inputs=[audio_input,input_language], outputs=audio_output)
|
|
|
|
|
431 |
|
432 |
|
433 |
iface.queue().launch(show_error=True,debug=True)
|
|
|
15 |
dotenv.load_dotenv()
|
16 |
|
17 |
seamless_client = Client("facebook/seamless_m4t")
|
18 |
+
HuggingFace_Token = os.getenv("HuggingFace_Token")
|
19 |
|
20 |
+
def check_hallucination(assertion,citation):
|
21 |
+
API_URL = "https://api-inference.huggingface.co/models/vectara/hallucination_evaluation_model"
|
22 |
+
headers = {"Authorization": f"Bearer {HuggingFace_Token}"}
|
23 |
+
payload = {"inputs" : f"{assertion} [SEP] {citation}"}
|
24 |
+
|
25 |
+
response = requests.post(API_URL, headers=headers, json=payload,timeout=120)
|
26 |
+
output = response.json()
|
27 |
+
output = output[0][0]["score"]
|
28 |
+
|
29 |
+
return f"**hullicination score:** {output}"
|
30 |
|
31 |
|
32 |
|
|
|
436 |
audio_input = gr.Audio(label="speak",type="filepath",sources="microphone")
|
437 |
audio_output = gr.Markdown(label="output text")
|
438 |
audio_button = gr.Button("process audio")
|
439 |
+
with gr.Tab("hallucination check"):
|
440 |
+
assertion = gr.Textbox(label="assertion")
|
441 |
+
citation = gr.Textbox(label="citation text")
|
442 |
+
hullucination_output = gr.Markdown(label="output text")
|
443 |
+
audio_button = gr.Button("check hallucination")
|
444 |
text_button.click(process_and_query, inputs=text_input, outputs=text_output)
|
445 |
image_button.click(process_image, inputs=image_input, outputs=image_output)
|
446 |
audio_button.click(process_speech, inputs=[audio_input,input_language], outputs=audio_output)
|
447 |
+
audio_button.click(check_hallucination,inputs=[assertion,citation],outputs=hullucination_output)
|
448 |
+
|
449 |
|
450 |
|
451 |
iface.queue().launch(show_error=True,debug=True)
|