Views
No views yet
1from transformers import pipeline
2import torch
3
4model_id = "sequelbox/Qwen3-14B-UML-Generator"
5
6pipe = pipeline(
7 "text-generation",
8 model=model_id,
9 torch_dtype="auto",
10 device_map="auto",
11)
12
13prompt = "Generate a UML class diagram modeling a feature toggle framework. A FeatureToggleManager provides an isEnabled(featureName) method. The manager's implementation must be designed using the Strategy pattern, where it delegates the actual check to an IFeatureToggleStrategy interface. This allows the framework to support different backends (e.g., ConfigFileStrategy, DatabaseStrategy, CloudConfigStrategy) for storing the toggle states."
14#prompt = "Generate a UML component diagram for a CAN bus interface, clearly separating the hardware and software. Show a Vehicle_Control_App component requiring an I_CAN_Message interface (e.g., sendMessage(id, data)). This interface is provided by a CAN_HAL component. The CAN_HAL component, in turn, requires an I_CAN_Registers interface, which is provided by the <<hardware>>CAN_Controller component. The <<hardware>>CAN_Controller is then connected to a <<device>>CAN_Transceiver."
15#prompt = "Generate a UML composite structure diagram for a HospitalInformationSystem component. This system must show its internal parts (subsystems): PatientAdmissions, ElectronicHealthRecord (EHR), Billing, and Pharmacy. PatientAdmissions must be connected to EHR and Billing. EHR must be connected to Pharmacy and Billing. The EHR part must also expose an external port HL7_Interface for communicating with external lab systems."
16#prompt = "Generate a UML State Machine Diagram for an avionics Flight Mode Annunciator. The FMA must have two orthogonal regions: Autothrottle (States: Off, ARM, N1, SPEED) and Autopilot (States: Off, FD (Flight Director), CMD). The Autopilot region must also be a composite state with nested orthogonal regions for LateralMode (States: LNAV, HDG_SEL) and VerticalMode (States: LVNAV, ALT_HOLD). A TakeoffGoAround (TOGA) event must force-transition all regions to their respective armed/takeoff states."
17
18messages = [
19 {"role": "user", "content": prompt},
20]
21
22outputs = pipe(
23 messages,
24 max_new_tokens=16000,
25)
26print(outputs[0]["generated_text"][-1])
27