premuim89 commited on
Commit
bb30f5c
·
verified ·
1 Parent(s): 6dab2c9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # Pretrained model load karo (direct original repo se)
5
+ model_name = "username-of-original-author/uitars1.5" # ← Replace karo actual model path se
6
+ generator = pipeline("text-generation", model=model_name)
7
+
8
+ # Text generate karne ka function
9
+ def generate_text(prompt):
10
+ result = generator(prompt, max_new_tokens=100, do_sample=True)
11
+ return result[0]["generated_text"]
12
+
13
+ # Gradio Interface
14
+ gr.Interface(
15
+ fn=generate_text,
16
+ inputs=gr.Textbox(lines=2, placeholder="Enter your prompt..."),
17
+ outputs="text",
18
+ title="UITARS 1.5 Text Generator"
19
+ ).launch()