Hodely commited on
Commit
a4aba75
verified
1 Parent(s): f1bda60

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +52 -0
app.py ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
4
+ from peft import PeftModel
5
+
6
+ BASE_MODEL = "Qwen/Qwen3-4B-Instruct-2507"
7
+ ADAPTER_MODEL = "Hodely/AmInside-Qwen3-4B"
8
+
9
+ tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, trust_remote_code=True)
10
+
11
+ base_model = AutoModelForCausalLM.from_pretrained(
12
+ BASE_MODEL,
13
+ device_map="auto",
14
+ torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
15
+ trust_remote_code=True
16
+ )
17
+
18
+ model = PeftModel.from_pretrained(base_model, ADAPTER_MODEL)
19
+
20
+ generator = pipeline(
21
+ "text-generation",
22
+ model=model,
23
+ tokenizer=tokenizer
24
+ )
25
+
26
+ def chat(message, history):
27
+ prompt = f"""<|im_start|>system
28
+ Eres AmSide, una inteligencia artificial basada en AmInSide1.0, creada por HodelyGil. Te adaptas al prompt, a webs, textos o cualquier informaci贸n que el usuario te d茅. Ayudas a estudiar, crear, programar, explicar y resolver tareas generales.<|im_end|>
29
+ <|im_start|>user
30
+ {message}<|im_end|>
31
+ <|im_start|>assistant
32
+ """
33
+
34
+ result = generator(
35
+ prompt,
36
+ max_new_tokens=220,
37
+ temperature=0.7,
38
+ do_sample=True,
39
+ repetition_penalty=1.08
40
+ )[0]["generated_text"]
41
+
42
+ answer = result.split("<|im_start|>assistant")[-1]
43
+ answer = answer.replace("<|im_end|>", "").strip()
44
+ return answer
45
+
46
+ demo = gr.ChatInterface(
47
+ fn=chat,
48
+ title="AmSide",
49
+ description="AmSide 路 Modelo AmInSide1.0 creado por HodelyGil"
50
+ )
51
+
52
+ demo.launch()