File size: 532 Bytes
71e2c65
1112e46
 
95bfa0d
71e2c65
59284b1
4c66f77
1112e46
 
 
4c66f77
 
71e2c65
95bfa0d
 
 
 
71e2c65
9d92f1e
71e2c65
a21e182
71e2c65
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import gradio as gr
import os

from loaddataset import ExtractRagBenchData
from model import generate_response
from huggingface_hub import login
from huggingface_hub import whoami
from huggingface_hub import dataset_info


hf_token = os.getenv("HF_TOKEN")
login(hf_token)

rag_extracted_data = ExtractRagBenchData()

rag_extracted_data.head(5)

def chatbot(prompt):
    return whoami()

iface = gr.Interface(fn=chatbot, inputs="text", outputs="text", title="Capstone Project Group 10")

if __name__ == "__main__":
    iface.launch()