Spaces:
Sleeping
Sleeping
File size: 2,367 Bytes
321de10 f8dc88c d0bd435 321de10 d0bd435 321de10 d0bd435 321de10 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import gradio as gr
from torchvision import transforms
import torch
from inference import run_inference
#####
description_zero_shot_training = """ ### Chat with Phi-2 """
# Description
title = "<center><strong><font size='8'>📎 Phi-2 Chat (Trained on OpenAssistant/oasst1 Dataset with QLoRA) 📎</font></strong></center>"
text_input = gr.Text(label="Enter text")
text_input2 = gr.Text(label="Generated Response")
css = "h1 { text-align: center } .about { text-align: justify; padding-left: 10%; padding-right: 10%; }"
with gr.Blocks(css=css, title='Play with CLIP') as demo:
with gr.Row():
with gr.Column(scale=1):
# Title
gr.Markdown(title)
with gr.Tab("chat_with_phi2"):
# Images
with gr.Row(variant="panel"):
with gr.Column(scale=1):
text_input.render()
with gr.Column(scale=1):
text_input2.render()
# Submit & Clear
with gr.Row():
with gr.Column():
run_chat_with_phi2_button = gr.Button("chat with phi2", variant='primary')
clear_btn_text_to_image = gr.Button("Clear", variant="secondary")
gr.Markdown(description_zero_shot_training)
gr.Examples(examples = ["What is Large Language models ?", "Can you write a short introduction about the relevance of the term monopsony in economics? Please use examples related to potential monopsonies in the labour market and cite relevant research.", "I want to start doing astrophotography as a hobby, any suggestions what could i do?"],
inputs=[text_input],
outputs=text_input2,
fn=run_inference,
cache_examples=True,
examples_per_page=4)
run_chat_with_phi2_button.click(run_inference,
inputs=[
text_input,
],
outputs=text_input2)
#######################################################################################################################
def clear():
return None, None
def clear_text():
return None, None, None
clear_btn_text_to_image.click(clear, outputs=[text_input, text_input2])
demo.queue()
demo.launch()
|